Alternatives and similar repositories for FlashAttention20

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

• haochengxi / Train_Transformers_with_INT4
  ☆147Updated last year
• ChenMnZ / PrefixQuant
  An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization
  ☆133Updated 3 months ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆187Updated 11 months ago
• OpenNLPLab / lightning-attention
  Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models
  ☆285Updated 2 months ago
• hahnyuan / RPTQ4LLM
  Reorder-based post-training quantization for large language model
  ☆190Updated last year
• hahnyuan / PB-LLM
  PB-LLM: Partially Binarized Large Language Models
  ☆152Updated last year
• nbasyl / LLM-FP4
  The official implementation of the EMNLP 2023 paper LLM-FP4
  ☆198Updated last year
• tridao / flash-attention-wheels
  ☆49Updated last year
• efeslab / Atom
  [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving
  ☆308Updated 10 months ago
• opengear-project / GEAR
  GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM
  ☆161Updated 10 months ago
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆190Updated 3 weeks ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆89Updated 2 weeks ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆158Updated last year
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆209Updated 8 months ago
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆248Updated 6 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆124Updated this week
• Guangxuan-Xiao / torch-int
  This repository contains integer operators on GPUs for PyTorch.
  ☆204Updated last year
• SqueezeAILab / KVQuant
  [NeurIPS 2024] KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization
  ☆348Updated 9 months ago
• stanford-futuredata / stk
  ☆104Updated 8 months ago
• casper-hansen / AutoAWQ_kernels
  ☆73Updated 5 months ago
• wejoncy / QLLM
  A general 2-8 bits quantization toolbox with GPTQ/AWQ/HQQ/VPTQ, and export to onnx/onnx-runtime easily.
  ☆169Updated last month
• kyegomez / FlashAttention20Triton
  Triton implementation of Flash Attention2.0
  ☆33Updated last year
• HandH1998 / QQQ
  QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.
  ☆121Updated last month
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆68Updated 2 months ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆195Updated this week
• thu-ml / low-bit-optimizers
  Low-bit optimizers for PyTorch
  ☆128Updated last year
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆265Updated 2 weeks ago
• shreyansh26 / FlashAttention-PyTorch
  Implementation of FlashAttention in PyTorch
  ☆146Updated 4 months ago
• FasterDecoding / TEAL
  ☆128Updated 3 months ago
• ruikangliu / FlatQuant
  Official PyTorch implementation of "FlatQuant: Flatness Matters for LLM Quantization"
  ☆127Updated last week