meta-pytorch / krakenLinks
Triton-based Symmetric Memory operators and examples
☆31Updated last week
Alternatives and similar repositories for kraken
Users that are interested in kraken are comparing it to the libraries listed below
Sorting:
- extensible collectives library in triton☆88Updated 6 months ago
- ☆90Updated 11 months ago
- DeeperGEMM: crazy optimized version☆71Updated 5 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆84Updated 2 weeks ago
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆111Updated last week
- ☆33Updated last week
- ring-attention experiments☆152Updated 11 months ago
- ☆39Updated 2 months ago
- A bunch of kernels that might make stuff slower 😉☆59Updated this week
- ☆64Updated 5 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆123Updated 4 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆230Updated last week
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆97Updated 3 months ago
- Effective transpose on Hopper GPU☆24Updated last month
- ☆50Updated 4 months ago
- Estimate MFU for DeepSeekV3☆24Updated 9 months ago
- How to ensure correctness and ship LLM generated kernels in PyTorch☆66Updated last week
- Boosting 4-bit inference kernels with 2:4 Sparsity☆82Updated last year
- Triton-based implementation of Sparse Mixture of Experts.☆242Updated this week
- ☆129Updated 4 months ago
- Collection of kernels written in Triton language☆157Updated 6 months ago
- Debug print operator for cudagraph debugging☆13Updated last year
- ☆43Updated 5 months ago
- Example of applying CUDA graphs to LLaMA-v2☆12Updated 2 years ago
- ☆72Updated 6 months ago
- ☆113Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆224Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆77Updated last year
- An experimental communicating attention kernel based on DeepEP.☆34Updated 2 months ago
- Applied AI experiments and examples for PyTorch☆296Updated last month