Alternatives and similar repositories for 2by4-pretrain

Users that are interested in 2by4-pretrain are comparing it to the libraries listed below

• mit-han-lab / x-attention
  [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆263Updated 6 months ago
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆256Updated 5 months ago
• alexzhang13 / flashattention2-custom-mask
  Triton implementation of FlashAttention2 that adds Custom Masks.
  ☆159Updated last year
• mit-han-lab / flash-moba
  ☆213Updated last month
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆109Updated 9 months ago
• mit-han-lab / Quest
  [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference
  ☆362Updated 6 months ago
• thu-ml / Jetfire-INT8Training
  ☆60Updated last year
• FasterDecoding / TEAL
  ☆157Updated 10 months ago
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆333Updated last month
• IST-DASLab / HALO
  HALO: Hadamard-Assisted Low-Precision Optimization and Training method for finetuning LLMs. 🚀 The official implementation of https://arx…
  ☆29Updated 10 months ago
• ChenMnZ / PrefixQuant
  An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization
  ☆170Updated last month
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆160Updated 2 months ago
• Dao-AILab / grouped-latent-attention
  ☆133Updated 7 months ago
• attention-survey / Efficient_Attention_Survey
  A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention
  ☆270Updated last month
• microsoft / SeerAttention
  SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs
  ☆182Updated 3 months ago
• XunhaoLai / native-sparse-attention-triton
  Efficient triton implementation of Native Sparse Attention.
  ☆257Updated 7 months ago
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆430Updated this week
• z-lab / sparselora
  [ICML 2025] SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity
  ☆65Updated 6 months ago
• shadowpa0327 / Palu
  [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection
  ☆151Updated 10 months ago
• thu-ml / 2by4-pretrain-acc-examples
  Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"
  ☆27Updated last year
• mit-han-lab / fouroversix
  Code for the paper “Four Over Six: More Accurate NVFP4 Quantization with Adaptive Block Scaling”
  ☆104Updated last week
• opengear-project / GEAR
  GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM
  ☆175Updated last year
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆415Updated 3 months ago
• osayamenja / FlashMoE
  Distributed MoE in a Single Kernel [NeurIPS '25]
  ☆174Updated this week
• Aaronhuang-778 / Mixture-Compressor-MoE
  [ICLR 2025] Mixture Compressor for Mixture-of-Experts LLMs Gains More
  ☆65Updated 11 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 month
• hahnyuan / ASVD4LLM
  Activation-aware Singular Value Decomposition for Compressing Large Language Models
  ☆83Updated last year
• svg-project / flash-kmeans
  Fast and memory-efficient exact kmeans
  ☆133Updated 2 months ago
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆225Updated 6 months ago
• HanGuo97 / log-linear-attention
  ☆263Updated 7 months ago