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
  ☆255Updated 4 months ago
• mit-han-lab / flash-moba
  ☆187Updated last week
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆108Updated 8 months ago
• Dao-AILab / grouped-latent-attention
  ☆132Updated 6 months ago
• thu-ml / Jetfire-INT8Training
  ☆60Updated last year
• z-lab / sparselora
  [ICML 2025] SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity
  ☆60Updated 4 months ago
• alexzhang13 / flashattention2-custom-mask
  Triton implementation of FlashAttention2 that adds Custom Masks.
  ☆151Updated last year
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆250Updated 3 months ago
• mit-han-lab / Quest
  [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference
  ☆353Updated 4 months ago
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆385Updated 9 months ago
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆311Updated 2 weeks ago
• XunhaoLai / native-sparse-attention-triton
  Efficient triton implementation of Native Sparse Attention.
  ☆250Updated 6 months ago
• FasterDecoding / TEAL
  ☆151Updated 9 months ago
• ChenMnZ / PrefixQuant
  An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization
  ☆166Updated this week
• IST-DASLab / HALO
  HALO: Hadamard-Assisted Low-Precision Optimization and Training method for finetuning LLMs. 🚀 The official implementation of https://arx…
  ☆29Updated 9 months ago
• thu-ml / 2by4-pretrain-acc-examples
  Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"
  ☆26Updated 11 months ago
• microsoft / SeerAttention
  SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs
  ☆173Updated 2 months ago
• osayamenja / FlashMoE
  Distributed MoE in a Single Kernel [NeurIPS '25]
  ☆145Updated 2 months ago
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆155Updated last month
• opengear-project / GEAR
  GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM
  ☆170Updated last year
• HanGuo97 / log-linear-attention
  ☆256Updated 5 months ago
• shadowpa0327 / Palu
  [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection
  ☆148Updated 9 months ago
• attention-survey / Efficient_Attention_Survey
  A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention
  ☆233Updated 3 months ago
• INT-FlashAttention2024 / INT-FlashAttention
  ☆83Updated 10 months ago
• mdy666 / Scalable-Flash-Native-Sparse-Attention
  ☆47Updated last month
• svg-project / flash-kmeans
  Fast and memory-efficient exact kmeans
  ☆126Updated 2 weeks ago
• Intelligent-Computing-Lab-Panda / GPTAQ
  Code implementation of GPTAQ (https://arxiv.org/abs/2504.02692)
  ☆76Updated 4 months ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆259Updated last month
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆131Updated 6 months ago
• sustcsonglin / linear-attention-and-beyond-slides
  ☆96Updated 9 months ago