Alternatives and similar repositories for flash-muon

Users that are interested in flash-muon are comparing it to the libraries listed below

• Dao-AILab / grouped-latent-attention
  ☆53Updated this week
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆127Updated this week
• FasterDecoding / TEAL
  ☆129Updated 3 months ago
• PiotrNawrot / sparse-frontier
  Official implementation of "The Sparse Frontier: Sparse Attention Trade-offs in Transformer LLMs"
  ☆32Updated last month
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆69Updated 11 months ago
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆146Updated 3 weeks ago
• PiotrNawrot / nano-sparse-attention
  The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.
  ☆62Updated 4 months ago
• XunhaoLai / native-sparse-attention-triton
  Efficient triton implementation of Native Sparse Attention.
  ☆155Updated last week
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆68Updated 3 months ago
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆41Updated last month
• dame-cell / Triformer
  Transformers components but in Triton
  ☆33Updated 3 weeks ago
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆43Updated 8 months ago
• IST-DASLab / Sparse-Marlin
  Boosting 4-bit inference kernels with 2:4 Sparsity
  ☆75Updated 8 months ago
• mgmalek / efficient_cross_entropy
  ☆108Updated last year
• sustcsonglin / linear-attention-and-beyond-slides
  ☆74Updated 3 months ago
• huggingface / kernels
  Load compute kernels from the Hub
  ☆139Updated this week
• Dao-AILab / gemm-cublas
  ☆20Updated 3 weeks ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆93Updated this week
• IST-DASLab / QuEST
  Work in progress.
  ☆67Updated this week
• frankxwang / dpo-prefix-sharing
  DPO, but faster 🚀
  ☆42Updated 5 months ago
• kyleliang919 / Super_Muon
  ☆56Updated 2 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• mit-han-lab / x-attention
  XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆158Updated 3 weeks ago
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆216Updated 6 months ago
• microsoft / AttentionEngine
  ☆70Updated 2 weeks ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆143Updated 7 months ago
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆75Updated last year
• JonasGeiping / linear_cross_entropy_loss
  A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
  ☆67Updated 10 months ago
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆203Updated 2 weeks ago
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆70Updated last year