Alternatives and similar repositories for FlashAttention20

Users that are interested in FlashAttention20 are comparing it to the libraries listed below

• tridao / flash-attention-wheels
  ☆50Updated last year
• Dao-AILab / grouped-latent-attention
  ☆114Updated 3 weeks ago
• ChenMnZ / PrefixQuant
  An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization
  ☆137Updated last month
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆99Updated 3 weeks ago
• thu-ml / low-bit-optimizers
  Low-bit optimizers for PyTorch
  ☆129Updated last year
• haochengxi / Train_Transformers_with_INT4
  ☆151Updated 2 years ago
• kyegomez / FlashAttention20Triton
  Triton implementation of Flash Attention2.0
  ☆35Updated last year
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆157Updated last year
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆210Updated last week
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆210Updated 10 months ago
• alexzhang13 / flashattention2-custom-mask
  Triton implementation of FlashAttention2 that adds Custom Masks.
  ☆120Updated 10 months ago
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆178Updated 2 weeks ago
• hahnyuan / PB-LLM
  PB-LLM: Partially Binarized Large Language Models
  ☆152Updated last year
• AniZpZ / AutoSmoothQuant
  An easy-to-use package for implementing SmoothQuant for LLMs
  ☆102Updated 2 months ago
• RobertCsordas / moe_attention
  Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"
  ☆98Updated 8 months ago
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆272Updated last year
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆136Updated this week
• FasterDecoding / TEAL
  ☆130Updated 4 months ago
• efeslab / Atom
  [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving
  ☆310Updated 11 months ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆201Updated last year
• OpenNLPLab / lightning-attention
  Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models
  ☆313Updated 4 months ago
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆131Updated last week
• ByteDance-Seed / decoupleQ
  A quantization algorithm for LLM
  ☆141Updated last year
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆252Updated 7 months ago
• hahnyuan / RPTQ4LLM
  Reorder-based post-training quantization for large language model
  ☆191Updated 2 years ago
• SqueezeBits / QUICK
  QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference
  ☆118Updated last year
• bytedance / AffineQuant
  Official implementation of the ICLR 2024 paper AffineQuant
  ☆26Updated last year
• DD-DuDa / BitDistiller
  [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.
  ☆115Updated last year
• shreyansh26 / FlashAttention-PyTorch
  Implementation of FlashAttention in PyTorch
  ☆151Updated 5 months ago
• InternLM / turbomind
  ☆86Updated 3 months ago