Alternatives and similar repositories for MHA2MLA

Users that are interested in MHA2MLA are comparing it to the libraries listed below

• QwenLM / ParScale
  Parallel Scaling Law for Language Model — Beyond Parameter and Inference Time Scaling
  ☆467Updated 8 months ago
• MuLabPKU / TransMLA
  TransMLA: Multi-Head Latent Attention Is All You Need (NeurIPS 2025 Spotlight)
  ☆426Updated 3 months ago
• step-law / steplaw
  ☆209Updated 2 months ago
• SkyworkAI / Skywork-MoE
  Skywork-MoE: A Deep Dive into Training Techniques for Mixture-of-Experts Language Models
  ☆138Updated last year
• mit-han-lab / duo-attention
  [ICLR 2025] DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads
  ☆519Updated 11 months ago
• yaof20 / Flash-RL
  Implementation for FP8/INT8 Rollout for RL training without performence drop.
  ☆283Updated 2 months ago
• thu-ml / ReMoE
  [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.
  ☆104Updated last year
• ZihanWang314 / CoE
  Chain of Experts (CoE) enables communication between experts within Mixture-of-Experts (MoE) models
  ☆227Updated 2 months ago
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆257Updated 5 months ago
• XunhaoLai / native-sparse-attention-triton
  Efficient triton implementation of Native Sparse Attention.
  ☆260Updated 7 months ago
• bigai-nlco / TokenSwift
  [ICML 2025] |TokenSwift: Lossless Acceleration of Ultra Long Sequence Generation
  ☆120Updated 8 months ago
• stepfun-ai / Step3
  ☆444Updated 5 months ago
• OpenSparseLLMs / Linear-MoE
  ☆127Updated 7 months ago
• hao-ai-lab / Dynasor
  [NeurIPS 2025] Simple extension on vLLM to help you speed up reasoning model without training.
  ☆217Updated 7 months ago
• microsoft / SeerAttention
  SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs
  ☆187Updated 3 months ago
• thu-nics / MoA
  [CoLM'25] The official implementation of the paper <MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression>
  ☆153Updated last week
• astramind-ai / Mixture-of-depths
  Unofficial implementation for the paper "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"
  ☆176Updated last year
• weigao266 / Awesome-Efficient-Arch
  Speed Always Wins: A Survey on Efficient Architectures for Large Language Models
  ☆389Updated 2 months ago
• z-lab / dflash
  Block Diffusion for Ultra-Fast Speculative Decoding
  ☆313Updated 2 weeks ago
• mdy666 / Qwen-Native-Sparse-Attention
  qwen-nsa
  ☆87Updated 3 months ago
• RyanLiu112 / compute-optimal-tts
  Official codebase for "Can 1B LLM Surpass 405B LLM? Rethinking Compute-Optimal Test-Time Scaling".
  ☆280Updated 11 months ago
• OpenNLPLab / lightning-attention
  Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models
  ☆339Updated 10 months ago
• NVIDIA / Star-Attention
  Efficient LLM Inference over Long Sequences
  ☆393Updated 6 months ago
• shangshang-wang / Tina
  Tina: Tiny Reasoning Models via LoRA
  ☆314Updated 3 months ago
• NVlabs / MaskLLM
  [NeurIPS 24 Spotlight] MaskLLM: Learnable Semi-structured Sparsity for Large Language Models
  ☆183Updated last year
• Tencent / llm.hunyuan.T1
  ☆83Updated 9 months ago
• OpenBMB / infllmv2_cuda_impl
  ☆81Updated last month
• LCLM-Horizon / A-Comprehensive-Survey-For-Long-Context-Language-Modeling
  A Comprehensive Survey on Long Context Language Modeling
  ☆215Updated last month
• FlagAI-Open / OpenSeek
  OpenSeek aims to unite the global open source community to drive collaborative innovation in algorithms, data and systems to develop next…
  ☆240Updated 3 weeks ago
• thunlp / JustRL
  ☆196Updated 3 weeks ago