Alternatives and similar repositories for Tiny-Megatron

Users that are interested in Tiny-Megatron are comparing it to the libraries listed below

• mdy666 / Qwen-Native-Sparse-Attention
  qwen-nsa
  ☆85Updated 2 months ago
• attention-survey / Efficient_Attention_Survey
  A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention
  ☆265Updated 3 weeks ago
• mit-han-lab / flash-moba
  ☆210Updated last month
• pprp / Awesome-Efficient-MoE
  Efficient Mixture of Experts for LLM Paper List
  ☆153Updated 2 months ago
• ifromeast / AI_analysis
  analyse problems of AI with Math and Code
  ☆27Updated 4 months ago
• ThisisBillhe / ZipCache
  [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification
  ☆31Updated 8 months ago
• yaof20 / Flash-RL
  Implementation for FP8/INT8 Rollout for RL training without performence drop.
  ☆281Updated last month
• liangyuwang / Tiny-DeepSpeed
  Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library
  ☆48Updated 4 months ago
• dhcode-cpp / NSA-pytorch
  DeepSeek Native Sparse Attention pytorch implementation
  ☆110Updated last week
• smart-lty / ParallelSpeculativeDecoding
  [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length
  ☆139Updated this week
• hao-ai-lab / Awesome-Video-Attention
  A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cach…
  ☆52Updated last month
• UNITES-Lab / Occult
  [ICML‘25] Official code for paper "Occult: Optimizing Collaborative Communication across Experts for Accelerated Parallel MoE Training an…
  ☆11Updated 8 months ago
• mit-han-lab / fastrl
  [ASPLOS'26] Taming the Long-Tail: Efficient Reasoning RL Training with Adaptive Drafter
  ☆114Updated 3 weeks ago
• maomaocun / dLLM-cache
  Official PyTorch implementation of the paper "dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching" (dLLM-Cache…
  ☆189Updated last month
• OpenMOSS / DiRL
  ☆95Updated 2 weeks ago
• mdy666 / mdy_triton
  ☆149Updated 5 months ago
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆159Updated 2 months ago
• FFY0 / AdaKV
  The Official Implementation of Ada-KV [NeurIPS 2025]
  ☆120Updated last month
• Gaffey / ExCP
  Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".
  ☆47Updated last year
• tsinghua-ideal / Twilight
  [NeurIPS'25 Spotlight] Adaptive Attention Sparsity with Hierarchical Top-p Pruning
  ☆79Updated 3 weeks ago
• sail-sg / LongSpec
  LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification
  ☆69Updated 5 months ago
• InternLM / Awesome-LLM-Training-System
  ☆45Updated last year
• thu-nics / MoA
  [CoLM'25] The official implementation of the paper <MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression>
  ☆154Updated last month
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆109Updated 9 months ago
• zhuzilin / flash-attention-with-sink
  ☆39Updated 4 months ago
• Dominic789654 / LongGenBench
  Source code for the paper "LongGenBench: Long-context Generation Benchmark"
  ☆24Updated last year
• OpenBMB / infllmv2_cuda_impl
  ☆78Updated 3 weeks ago
• mit-han-lab / x-attention
  [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆261Updated 5 months ago
• OpenSparseLLMs / Linear-MoE
  ☆126Updated 6 months ago
• ruipeterpan / specreason
  PoC for "SpecReason: Fast and Accurate Inference-Time Compute via Speculative Reasoning" [NeurIPS '25]
  ☆60Updated 2 months ago