Alternatives and similar repositories for xformers:

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

• xdit-project / DistVAE
  A parallelism VAE avoids OOM for high resolution image generation
  ☆50Updated last week
• Repeerc / flash-attention-v2-RDNA3-minimal
  a simple Flash Attention v2 implementation with ROCM (RDNA3 GPU, roc wmma), mainly used for stable diffusion(ComfyUI) in Windows ZLUDA en…
  ☆35Updated 5 months ago
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆152Updated this week
• huggingface / optimum-amd
  AMD related optimizations for transformer models
  ☆64Updated 2 months ago
• ROCm / TransformerEngine
  ☆15Updated this week
• HanGuo97 / flute
  Fast Matrix Multiplications for Lookup Table-Quantized LLMs
  ☆221Updated last week
• mit-han-lab / patch_conv
  Patch convolution to avoid large GPU memory usage of Conv2D
  ☆82Updated last week
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆37Updated 6 months ago
• Cornell-RelaxML / qtip
  ☆100Updated last month
• thu-nics / DiTFastAttn
  ☆131Updated 2 weeks ago
• xdit-project / DiTCacheAnalysis
  An auxiliary project analysis of the characteristics of KV in DiT Attention.
  ☆23Updated 2 months ago
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆232Updated 3 months ago
• chengzeyi / piflux
  (WIP) Parallel inference for black-forest-labs' FLUX model.
  ☆17Updated 2 months ago
• ROCm / bitsandbytes
  8-bit CUDA functions for PyTorch
  ☆42Updated 2 weeks ago
• KONAKONA666 / q8_kernels
  ☆57Updated last month
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆98Updated 3 months ago
• NVlabs / COAT
  ☆55Updated 3 weeks ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆86Updated this week
• aredden / torch-cublas-hgemm
  PyTorch half precision gemm lib w/ fused optional bias + optional relu/gelu
  ☆48Updated last month
• Infini-AI-Lab / MagicDec
  [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding
  ☆105Updated last month
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆99Updated 4 months ago
• mit-han-lab / deepcompressor
  Model Compression Toolbox for Large Language Models and Diffusion Models
  ☆309Updated last month
• mobiusml / gemlite
  Fast low-bit matmul kernels in Triton
  ☆199Updated last week
• chengzeyi / ParaAttention
  Context parallel attention that accelerates DiT model inference with dynamic caching
  ☆165Updated this week
• InternLM / turbomind
  ☆59Updated last month
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆94Updated 6 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆61Updated 2 months ago
• DefTruth / ffpa-attn-mma
  📚[WIP] FFPA: Yet antother Faster Flash Prefill Attention with O(1)⚡️GPU SRAM complexity for headdim > 256, 1.8x~3x↑🎉faster vs SDPA EA.
  ☆73Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆50Updated this week