mit-han-lab / Block-Sparse-Attention
A sparse attention kernel supporting mix sparse patterns
☆58Updated last month
Related projects ⓘ
Alternatives and complementary repositories for Block-Sparse-Attention
- 16-fold memory access reduction with nearly no loss☆58Updated last week
- Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization☆66Updated last week
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆203Updated 3 weeks ago
- ☆96Updated 2 months ago
- An algorithm for static activation quantization of LLMs☆79Updated 2 weeks ago
- Quantized Attention on GPU☆31Updated this week
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆149Updated 4 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆53Updated 3 weeks ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆278Updated 4 months ago
- ☆23Updated 4 months ago
- Patch convolution to avoid large GPU memory usage of Conv2D☆79Updated 5 months ago
- Code Repository of Evaluating Quantized Large Language Models☆103Updated 2 months ago
- ☆42Updated 7 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆57Updated 5 months ago
- SKVQ: Sliding-window Key and Value Cache Quantization for Large Language Models☆13Updated last month
- Code for Palu: Compressing KV-Cache with Low-Rank Projection☆57Updated last week
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆33Updated 4 months ago
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆89Updated last month
- ☆88Updated 2 months ago
- [ICML 2024 Oral] Any-Precision LLM: Low-Cost Deployment of Multiple, Different-Sized LLMs☆83Updated 3 months ago
- ☆47Updated 2 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆51Updated 2 months ago
- ☆79Updated 2 months ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆211Updated 3 weeks ago
- Model Compression Toolbox for Large Language Models and Diffusion Models☆233Updated 2 weeks ago
- This repository contains integer operators on GPUs for PyTorch.☆184Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆68Updated 4 months ago
- ☆24Updated 7 months ago
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆85Updated 6 months ago
- Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆79Updated this week