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

• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆71Updated 5 months ago
• flashinfer-ai / cutlass-viz
  ☆64Updated 5 months ago
• tile-ai / AttentionEngine
  ☆50Updated 4 months ago
• flashinfer-ai / debug-print
  Debug print operator for cudagraph debugging
  ☆13Updated last year
• KuangjuX / AttnLink
  An experimental communicating attention kernel based on DeepEP.
  ☆34Updated 2 months ago
• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆44Updated 10 months ago
• DeepLink-org / DLSlime
  DLSlime: Flexible & Efficient Heterogeneous Transfer Toolkit
  ☆66Updated last week
• ByteDance-Seed / cudaLLM
  ☆115Updated last month
• cherichy / tilecute
  ☆30Updated 3 months ago
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆43Updated 3 months ago
• tile-ai / TileOPs
  ☆53Updated last week
• feifeibear / DPSKV3MFU
  Estimate MFU for DeepSeekV3
  ☆24Updated 9 months ago
• HanGuo97 / hilt
  ☆33Updated last week
• DD-DuDa / BitDecoding
  A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.
  ☆60Updated last month
• microsoft / AttentionEngine
  ☆99Updated 4 months ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆84Updated 2 weeks ago
• tile-ai / tilescale
  Tile-based language built for AI computation across all scales
  ☆66Updated last week
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆142Updated 4 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆123Updated 4 months ago
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆105Updated 6 months ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆50Updated last year
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆96Updated 9 months ago
• wu-kan / wuk_cupti_wrapper
  a simple API to use CUPTI
  ☆11Updated last month
• KuangjuX / NVSHMEM-Tutorial
  NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer
  ☆132Updated 2 weeks ago
• fzyzcjy / torch_utils
  Utility scripts for PyTorch (e.g. Memory profiler that understands more low-level allocations such as NCCL)
  ☆55Updated 3 weeks ago
• INT-FlashAttention2024 / INT-FlashAttention
  ☆82Updated 8 months ago
• cat538 / SKVQ
  [COLM 2024] SKVQ: Sliding-window Key and Value Cache Quantization for Large Language Models
  ☆24Updated last year
• AlibabaPAI / FLASHNN
  ☆98Updated last year
• microsoft / chunk-attention
  ☆78Updated 5 months ago
• PKU-SEC-Lab / HybriMoE
  [DAC'25] Official implement of "HybriMoE: Hybrid CPU-GPU Scheduling and Cache Management for Efficient MoE Inference"
  ☆72Updated 3 months ago