Alternatives and similar repositories for 2by4-pretrain

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

• thu-ml / Jetfire-INT8Training
  ☆60Updated last year
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆258Updated 5 months ago
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆109Updated 10 months ago
• alexzhang13 / flashattention2-custom-mask
  Triton implementation of FlashAttention2 that adds Custom Masks.
  ☆163Updated last year
• mit-han-lab / flash-moba
  ☆220Updated 2 months ago
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆160Updated 3 months ago
• mit-han-lab / Quest
  [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference
  ☆368Updated 6 months ago
• mit-han-lab / x-attention
  [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆267Updated 6 months ago
• ChenMnZ / PrefixQuant
  An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization
  ☆172Updated 2 months ago
• FasterDecoding / TEAL
  ☆158Updated 11 months ago
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆341Updated 2 months ago
• z-lab / sparselora
  [ICML 2025] SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity
  ☆69Updated 6 months ago
• attention-survey / Efficient_Attention_Survey
  A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention
  ☆278Updated 2 months ago
• mit-han-lab / fouroversix
  Code for the paper “Four Over Six: More Accurate NVFP4 Quantization with Adaptive Block Scaling”
  ☆116Updated this week
• XunhaoLai / native-sparse-attention-triton
  Efficient triton implementation of Native Sparse Attention.
  ☆262Updated 8 months ago
• IST-DASLab / HALO
  HALO: Hadamard-Assisted Low-Precision Optimization and Training method for finetuning LLMs. 🚀 The official implementation of https://arx…
  ☆29Updated 11 months ago
• Dao-AILab / grouped-latent-attention
  ☆132Updated 8 months ago
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆446Updated 2 weeks ago
• microsoft / SeerAttention
  SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs
  ☆188Updated 4 months ago
• opengear-project / GEAR
  GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM
  ☆176Updated last year
• thu-ml / 2by4-pretrain-acc-examples
  Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"
  ☆27Updated last year
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆279Updated 3 months ago
• shadowpa0327 / Palu
  [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection
  ☆153Updated 11 months ago
• mdy666 / Scalable-Flash-Native-Sparse-Attention
  ☆47Updated last month
• sustcsonglin / linear-attention-and-beyond-slides
  ☆104Updated 11 months ago
• Hsu1023 / DuQuant
  [NeurIPS 2024 Oral🔥] DuQuant: Distributing Outliers via Dual Transformation Makes Stronger Quantized LLMs.
  ☆180Updated last year
• sustcsonglin / fla-tilelang
  ☆35Updated 10 months ago
• osayamenja / FlashMoE
  Distributed MoE in a Single Kernel [NeurIPS '25]
  ☆188Updated last week
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆229Updated 7 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆141Updated 8 months ago