sandyresearch / chipmunkLinks
🎬 3.7× faster video generation E2E 🖼️ 1.6× faster image generation E2E ⚡ ColumnSparseAttn 9.3× vs FlashAttn‑3 💨 ColumnSparseGEMM 2.5× vs cuBLAS
☆74Updated 2 weeks ago
Alternatives and similar repositories for chipmunk
Users that are interested in chipmunk are comparing it to the libraries listed below
Sorting:
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆31Updated 7 months ago
- ☆71Updated 6 months ago
- Patch convolution to avoid large GPU memory usage of Conv2D☆90Updated 5 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆138Updated last month
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆191Updated last week
- A parallelism VAE avoids OOM for high resolution image generation☆67Updated 5 months ago
- ☆116Updated last month
- Quantized Attention on GPU☆44Updated 7 months ago
- (WIP) Parallel inference for black-forest-labs' FLUX model.☆19Updated 7 months ago
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆47Updated 11 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆89Updated 2 weeks ago
- [ICML 2025] SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity☆33Updated last week
- A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cach…☆28Updated this week
- A sparse attention kernel supporting mix sparse patterns☆249Updated 5 months ago
- A WebUI for Side-by-Side Comparison of Media (Images/Videos) Across Multiple Folders☆24Updated 4 months ago
- Accelerate LLM preference tuning via prefix sharing with a single line of code☆42Updated last week
- ☆222Updated last month
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆118Updated last month
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆117Updated last year
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆184Updated this week
- ☆136Updated 5 months ago
- The evaluation framework for training-free sparse attention in LLMs☆82Updated 3 weeks ago
- HALO: Hadamard-Assisted Low-Precision Optimization and Training method for finetuning LLMs. 🚀 The official implementation of https://arx…☆17Updated 4 months ago
- ☆74Updated last month
- A Quirky Assortment of CuTe Kernels☆281Updated this week
- ☆49Updated last month
- Efficient triton implementation of Native Sparse Attention.☆175Updated last month
- ☆77Updated 5 months ago
- 🤗CacheDiT: A Training-free and Easy-to-use Cache Acceleration Toolbox for Diffusion Transformers🔥☆99Updated this week
- ☆170Updated 6 months ago