tilde-research / nsa-implLinks
An efficient implementation of the NSA (Native Sparse Attention) kernel
☆72Updated this week
Alternatives and similar repositories for nsa-impl
Users that are interested in nsa-impl are comparing it to the libraries listed below
Sorting:
- ☆114Updated 3 weeks ago
- The official implementation for Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free☆44Updated last month
- ☆21Updated 3 months ago
- ☆19Updated 6 months ago
- ☆51Updated 3 months ago
- ☆76Updated 4 months ago
- Accelerate LLM preference tuning via prefix sharing with a single line of code☆41Updated last month
- Stick-breaking attention☆57Updated last week
- The evaluation framework for training-free sparse attention in LLMs☆69Updated last week
- XAttention: Block Sparse Attention with Antidiagonal Scoring☆166Updated this week
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆46Updated 8 months ago
- Efficient triton implementation of Native Sparse Attention.☆168Updated last month
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆113Updated last month
- Estimate MFU for DeepSeekV3☆24Updated 5 months ago
- Transformers components but in Triton☆34Updated last month
- 16-fold memory access reduction with nearly no loss☆99Updated 3 months ago
- ☆104Updated 3 weeks ago
- Beyond KV Caching: Shared Attention for Efficient LLMs☆19Updated 11 months ago
- SQUEEZED ATTENTION: Accelerating Long Prompt LLM Inference☆47Updated 7 months ago
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆39Updated last year
- LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification☆54Updated 3 months ago
- 🔥 A minimal training framework for scaling FLA models☆178Updated 2 weeks ago
- Odysseus: Playground of LLM Sequence Parallelism☆70Updated last year
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆88Updated last month
- ☆85Updated 2 months ago
- Fast and memory-efficient exact attention☆68Updated 3 months ago
- "Found in the Middle: How Language Models Use Long Contexts Better via Plug-and-Play Positional Encoding" Zhenyu Zhang, Runjin Chen, Shiw…☆29Updated last year
- Quantized Attention on GPU☆44Updated 7 months ago
- ☆130Updated 4 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆131Updated last week