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
• mit-han-lab / flash-moba
  ☆207Updated last month
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆252Updated 4 months ago
• mit-han-lab / x-attention
  [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆256Updated 5 months ago
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆109Updated 8 months ago
• Dao-AILab / grouped-latent-attention
  ☆132Updated 6 months ago
• mit-han-lab / Quest
  [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference
  ☆357Updated 5 months ago
• 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
• osayamenja / FlashMoE
  Distributed MoE in a Single Kernel [NeurIPS '25]
  ☆155Updated this week
• z-lab / sparselora
  [ICML 2025] SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity
  ☆64Updated 5 months ago
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆321Updated last month
• FasterDecoding / TEAL
  ☆155Updated 10 months ago
• alexzhang13 / flashattention2-custom-mask
  Triton implementation of FlashAttention2 that adds Custom Masks.
  ☆155Updated last year
• opengear-project / GEAR
  GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM
  ☆172Updated last year
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆158Updated 2 months ago
• ChenMnZ / PrefixQuant
  An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization
  ☆169Updated 3 weeks ago
• XunhaoLai / native-sparse-attention-triton
  Efficient triton implementation of Native Sparse Attention.
  ☆254Updated 6 months ago
• shadowpa0327 / Palu
  [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection
  ☆149Updated 10 months ago
• thu-ml / 2by4-pretrain-acc-examples
  Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"
  ☆27Updated last year
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆135Updated 6 months ago
• Dao-AILab / sonic-moe
  Accelerating MoE with IO and Tile-aware Optimizations
  ☆351Updated this week
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆406Updated 10 months ago
• hahnyuan / ASVD4LLM
  Activation-aware Singular Value Decomposition for Compressing Large Language Models
  ☆82Updated last year
• attention-survey / Efficient_Attention_Survey
  A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention
  ☆252Updated 2 weeks ago
• Hsu1023 / DuQuant
  [NeurIPS 2024 Oral🔥] DuQuant: Distributing Outliers via Dual Transformation Makes Stronger Quantized LLMs.
  ☆177Updated last year
• thu-nics / qllm-eval
  Code Repository of Evaluating Quantized Large Language Models
  ☆137Updated last year
• microsoft / SeerAttention
  SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs
  ☆177Updated 2 months ago
• Aaronhuang-778 / Mixture-Compressor-MoE
  [ICLR 2025] Mixture Compressor for Mixture-of-Experts LLMs Gains More
  ☆63Updated 10 months ago
• mit-han-lab / fouroversix
  Code for the paper “Four Over Six: More Accurate NVFP4 Quantization with Adaptive Block Scaling”
  ☆78Updated this week
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆222Updated 6 months ago