Alternatives and similar repositories for NSA-pytorch

Users that are interested in NSA-pytorch are comparing it to the libraries listed below

• pprp / Awesome-Efficient-MoE
  Efficient Mixture of Experts for LLM Paper List
  ☆166Updated 4 months ago
• mdy666 / Qwen-Native-Sparse-Attention
  qwen-nsa
  ☆87Updated 3 months ago
• mdy666 / mdy_triton
  ☆152Updated 7 months ago
• microsoft / SeerAttention
  SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs
  ☆188Updated 4 months ago
• MuLabPKU / TransMLA
  TransMLA: Multi-Head Latent Attention Is All You Need (NeurIPS 2025 Spotlight)
  ☆429Updated 4 months ago
• JT-Ushio / MHA2MLA
  Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMs
  ☆203Updated 2 months ago
• mit-han-lab / x-attention
  [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆267Updated 7 months ago
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆160Updated 3 months ago
• smart-lty / ParallelSpeculativeDecoding
  [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length
  ☆147Updated last month
• qingkelab / qingketalk
  青稞Talk
  ☆190Updated 2 weeks ago
• OpenSparseLLMs / Linear-MoE
  ☆129Updated 8 months ago
• haiduo / Jakiro
  This repository is the official implementation of "Jakiro: Boosting Speculative Decoding with Decoupled Multi-Head via MoE"
  ☆36Updated 4 months ago
• stepfun-ai / Step3
  ☆449Updated 5 months ago
• yaof20 / Flash-RL
  Implementation for FP8/INT8 Rollout for RL training without performence drop.
  ☆289Updated 3 months ago
• InternLM / Awesome-LLM-Training-System
  ☆47Updated last year
• attention-survey / Efficient_Attention_Survey
  A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention
  ☆278Updated 2 months ago
• liangyuwang / Tiny-Megatron
  Tiny-Megatron, a minimalistic re-implementation of the Megatron library
  ☆21Updated 5 months ago
• madsys-dev / deepseekv2-profile
  ☆155Updated 11 months ago
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆258Updated 5 months ago
• Dao-AILab / grouped-latent-attention
  ☆131Updated 8 months ago
• liangyuwang / Tiny-DeepSpeed
  Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library
  ☆49Updated 5 months ago
• OpenBMB / infllmv2_cuda_impl
  ☆87Updated this week
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆110Updated 10 months ago
• OpenNLPLab / lightning-attention
  Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models
  ☆340Updated 11 months ago
• kyegomez / FlashAttention20Triton
  Triton implementation of Flash Attention2.0
  ☆49Updated 2 years ago
• FFY0 / AdaKV
  The Official Implementation of Ada-KV [NeurIPS 2025]
  ☆125Updated 2 months ago
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆453Updated 3 weeks ago
• rlite-project / RLite
  A lightweight reinforcement learning framework that integrates seamlessly into your codebase, empowering developers to focus on algorithm…
  ☆99Updated 5 months ago
• ifromeast / AI_analysis
  analyse problems of AI with Math and Code
  ☆27Updated 6 months ago
• Multiverse4FM / Multiverse
  ☆90Updated 7 months ago