mit-han-lab / Block-Sparse-Attention
A sparse attention kernel supporting mix sparse patterns
☆93Updated 3 months ago
Alternatives and similar repositories for Block-Sparse-Attention:
Users that are interested in Block-Sparse-Attention are comparing it to the libraries listed below
- 16-fold memory access reduction with nearly no loss☆63Updated 2 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆58Updated 2 months ago
- Quantized Attention on GPU☆34Updated last month
- ☆55Updated 3 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆234Updated last month
- Implement Flash Attention using Cute.☆65Updated last month
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆152Updated 6 months ago
- ☆134Updated 5 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆75Updated this week
- ☆30Updated 5 months ago
- ☆52Updated last month
- Fast Hadamard transform in CUDA, with a PyTorch interface☆132Updated 7 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆64Updated 7 months ago
- [ICML 2024 Oral] Any-Precision LLM: Low-Cost Deployment of Multiple, Different-Sized LLMs☆90Updated 3 weeks ago
- An algorithm for static activation quantization of LLMs☆107Updated this week
- Code for Palu: Compressing KV-Cache with Low-Rank Projection☆63Updated 2 months ago
- Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization☆90Updated 2 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆93Updated 6 months ago
- ☆40Updated last month
- Code Repository of Evaluating Quantized Large Language Models☆112Updated 4 months ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆230Updated 2 months ago
- ☆64Updated 2 months ago
- PyTorch library for cost-effective, fast and easy serving of MoE models.☆112Updated last month
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, achieve peak⚡️ performance☆43Updated this week
- Boosting 4-bit inference kernels with 2:4 Sparsity☆64Updated 4 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆290Updated 6 months ago
- ☆51Updated 9 months ago
- ☆107Updated 3 months ago
- ☆78Updated 4 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆195Updated last year