Alternatives and similar repositories for long-context-attention

Users that are interested in long-context-attention are comparing it to the libraries listed below

• zhuzilin / ring-flash-attention
  Ring attention implementation with flash attention
  ☆979Updated 5 months ago
• sgl-project / SpecForge
  Train speculative decoding models effortlessly and port them smoothly to SGLang serving.
  ☆683Updated this week
• fanshiqing / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆184Updated last month
• sail-sg / zero-bubble-pipeline-parallelism
  Zero Bubble Pipeline Parallelism
  ☆449Updated 9 months ago
• SandAI-org / MagiAttention
  A Distributed Attention Towards Linear Scalability for Ultra-Long Context, Heterogeneous Data Training
  ☆631Updated this week
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆453Updated 3 weeks ago
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆258Updated 6 months ago
• mit-han-lab / Quest
  [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference
  ☆372Updated 7 months ago
• flagos-ai / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆287Updated last year
• NVIDIA-NeMo / Megatron-Bridge
  Training library for Megatron-based models with bidirectional Hugging Face conversion capability
  ☆419Updated this week
• ByteDance-Seed / VeOmni
  VeOmni: Scaling Any Modality Model Training with Model-Centric Distributed Recipe Zoo
  ☆1,620Updated this week
• fla-org / native-sparse-attention
  🐳 Efficient Triton implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"
  ☆964Updated last week
• FMInference / H2O
  [NeurIPS'23] H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models.
  ☆502Updated last year
• flagos-ai / FlagScale
  FlagScale is a large model toolkit based on open-sourced projects.
  ☆474Updated last week
• yanring / Megatron-MoE-ModelZoo
  Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.
  ☆161Updated 3 weeks ago
• Dao-AILab / sonic-moe
  Accelerating MoE with IO and Tile-aware Optimizations
  ☆569Updated 3 weeks ago
• ByteDance-Seed / ByteCheckpoint
  ByteCheckpoint: An Unified Checkpointing Library for LFMs
  ☆269Updated last week
• kwai / Megatron-Kwai
  LLM training technologies developed by kwai
  ☆70Updated 3 weeks ago
• Azure / MS-AMP
  Microsoft Automatic Mixed Precision Library
  ☆635Updated 2 months ago
• madsys-dev / deepseekv2-profile
  ☆155Updated 11 months ago
• mit-han-lab / duo-attention
  [ICLR 2025] DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads
  ☆524Updated last year
• mit-han-lab / omniserve
  [MLSys'25] QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving; [MLSys'25] LServe: Efficient Long-sequence LLM Se…
  ☆810Updated 11 months ago
• hemingkx / Spec-Bench
  Spec-Bench: A Comprehensive Benchmark and Unified Evaluation Platform for Speculative Decoding (ACL 2024 Findings)
  ☆363Updated 9 months ago
• XunhaoLai / native-sparse-attention-triton
  Efficient triton implementation of Native Sparse Attention.
  ☆262Updated 8 months ago
• xlite-dev / Awesome-DiT-Inference
  📚A curated list of Awesome Diffusion Inference Papers with Codes: Sampling, Cache, Quantization, Parallelism, etc.🎉
  ☆518Updated 3 weeks ago
• feifeibear / LLMSpeculativeSampling
  Fast inference from large lauguage models via speculative decoding
  ☆886Updated last year
• volcengine / veScale
  Byted PyTorch Distributed for Hyperscale Training of LLMs and RLs
  ☆926Updated 2 months ago
• ISEEKYAN / mbridge
  Bridge Megatron-Core to Hugging Face/Reinforcement Learning
  ☆191Updated last week
• stepfun-ai / Step3
  ☆449Updated 6 months ago
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆222Updated last year