Alternatives and similar repositories for flash-attention-with-sink

Users that are interested in flash-attention-with-sink are comparing it to the libraries listed below

• flashinfer-ai / cutlass-viz
  ☆61Updated 4 months ago
• tile-ai / AttentionEngine
  ☆50Updated 3 months ago
• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆44Updated 9 months ago
• flashinfer-ai / debug-print
  Debug print operator for cudagraph debugging
  ☆13Updated last year
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆71Updated 3 months ago
• KuangjuX / AttnLink
  An experimental communicating attention kernel based on DeepEP.
  ☆34Updated 3 weeks ago
• ByteDance-Seed / cudaLLM
  ☆91Updated last week
• tile-ai / tilescale
  Tile-based language built for AI computation across all scales
  ☆46Updated this week
• feifeibear / DPSKV3MFU
  Estimate MFU for DeepSeekV3
  ☆24Updated 7 months ago
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆43Updated last month
• KuangjuX / NVSHMEM-Tutorial
  NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer
  ☆19Updated this week
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆133Updated 3 months ago
• Chtholly-Boss / swizzle
  A practical way of learning Swizzle
  ☆25Updated 6 months ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆47Updated last year
• xdit-project / DiTCacheAnalysis
  An auxiliary project analysis of the characteristics of KV in DiT Attention.
  ☆32Updated 8 months ago
• cat538 / SKVQ
  [COLM 2024] SKVQ: Sliding-window Key and Value Cache Quantization for Large Language Models
  ☆24Updated 10 months ago
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆104Updated 5 months ago
• microsoft / chunk-attention
  ☆78Updated 4 months ago
• xxyux / SpInfer
  SpInfer: Leveraging Low-Level Sparsity for Efficient Large Language Model Inference on GPUs
  ☆55Updated 5 months ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆81Updated last week
• DD-DuDa / BitDecoding
  A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.
  ☆57Updated 3 weeks ago
• Bruce-Lee-LY / decoding_attention
  Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.
  ☆40Updated 2 months ago
• Ascend / triton-ascend
  Triton adapter for Ascend. Mirror of https://gitee.com/ascend/triton-ascend
  ☆67Updated this week
• microsoft / AttentionEngine
  ☆77Updated 3 months ago
• HanGuo97 / cutedsl-utilities
  ☆23Updated this week
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆76Updated last year
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆95Updated 8 months ago
• tile-ai / TileOPs
  ☆44Updated 3 weeks ago
• meta-pytorch / kraken
  Triton-based Symmetric Memory operators and examples
  ☆23Updated this week
• DerrickYLJ / TidalDecode
  [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention
  ☆47Updated 3 weeks ago