dhcode-cpp / NSA-pytorchLinks
DeepSeek Native Sparse Attention pytorch implementation
☆83Updated 5 months ago
Alternatives and similar repositories for NSA-pytorch
Users that are interested in NSA-pytorch are comparing it to the libraries listed below
Sorting:
- TransMLA: Multi-Head Latent Attention Is All You Need☆335Updated 3 weeks ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆323Updated 5 months ago
- Official implementation of "Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decoding"☆320Updated this week
- qwen-nsa☆70Updated 3 months ago
- SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs☆141Updated last week
- ☆140Updated last month
- Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMs☆183Updated last month
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆213Updated last month
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆124Updated 2 months ago
- Efficient Mixture of Experts for LLM Paper List☆87Updated 7 months ago
- ☆113Updated 2 months ago
- ☆326Updated this week
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆102Updated 3 months ago
- [CoLM'25] The official implementation of the paper <MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression>☆141Updated 3 weeks ago
- A sparse attention kernel supporting mix sparse patterns☆262Updated 5 months ago
- ☆198Updated 3 months ago
- ☆42Updated last year
- ☆123Updated 2 months ago
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆38Updated last week
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training