mayank31398 / cute-kernels
A bunch of kernels that might make stuff slower 😉
☆29Updated this week
Alternatives and similar repositories for cute-kernels:
Users that are interested in cute-kernels are comparing it to the libraries listed below
- ☆73Updated 4 months ago
- extensible collectives library in triton☆84Updated 6 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆104Updated this week
- Cataloging released Triton kernels.☆212Updated 2 months ago
- ring-attention experiments☆128Updated 5 months ago
- Write a fast kernel and run it on Discord. See how you compare against the best!☆35Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆77Updated 5 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆102Updated 8 months ago
- Collection of kernels written in Triton language☆117Updated last month
- ☆102Updated 7 months ago
- ☆27Updated 2 months ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated 8 months ago
- ☆63Updated this week
- ☆13Updated 3 weeks ago
- Applied AI experiments and examples for PyTorch☆251Updated last week
- Experiment of using Tangent to autodiff triton☆78Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆71Updated 6 months ago
- ☆192Updated this week
- Fast low-bit matmul kernels in Triton☆272Updated this week
- a minimal cache manager for PagedAttention, on top of llama3.☆73Updated 7 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆74Updated this week
- Fast Hadamard transform in CUDA, with a PyTorch interface☆154Updated 10 months ago
- FlexAttention w/ FlashAttention3 Support☆26Updated 5 months ago
- ☆67Updated 2 months ago
- ☆60Updated 3 months ago
- ☆38Updated last year
- The simplest but fast implementation of matrix multiplication in CUDA.☆34Updated 8 months ago
- ☆193Updated 8 months ago
- Load compute kernels from the Hub☆107Updated this week
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆44Updated 8 months ago