microsoft / chunk-attentionLinks
☆78Updated 3 months ago
Alternatives and similar repositories for chunk-attention
Users that are interested in chunk-attention are comparing it to the libraries listed below
Sorting:
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆101Updated 3 months ago
- ☆54Updated last year
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆216Updated last year
- ☆60Updated 3 months ago
- 16-fold memory access reduction with nearly no loss☆102Updated 4 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆124Updated 2 months ago
- [NeurIPS 2024] Efficient LLM Scheduling by Learning to Rank☆51Updated 9 months ago
- ☆92Updated 4 months ago
- [NeurIPS 2024] The official implementation of "Kangaroo: Lossless Self-Speculative Decoding for Accelerating LLMs via Double Early Exitin…☆59Updated last year
- ☆75Updated 9 months ago
- ☆145Updated 5 months ago
- [DAC'25] Official implement of "HybriMoE: Hybrid CPU-GPU Scheduling and Cache Management for Efficient MoE Inference"☆64Updated last month
- A simple calculation for LLM MFU.☆42Updated 5 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆106Updated 2 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆165Updated last year
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆56Updated last week
- ☆96Updated 10 months ago
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆48Updated last year
- ☆49Updated 2 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆57Updated last year
- QAQ: Quality Adaptive Quantization for LLM KV Cache☆52Updated last year
- The Official Implementation of Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inference☆87Updated last month
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆307Updated 3 weeks ago
- SpInfer: Leveraging Low-Level Sparsity for Efficient Large Language Model Inference on GPUs☆50Updated 4 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆102Updated 3 months ago
- ☆109Updated 8 months ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆110Updated last year
- ATC23 AE☆46Updated 2 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆212Updated 11 months ago
- [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention☆44Updated 3 months ago