Alternatives and similar repositories for AttentionEngine

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

• ByteDance-Seed / cudaLLM
  ☆112Updated last month
• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆44Updated 10 months ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆50Updated last year
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆84Updated 2 weeks ago
• flashinfer-ai / cutlass-viz
  ☆64Updated 5 months ago
• tile-ai / AttentionEngine
  ☆50Updated 4 months ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆77Updated last year
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆43Updated 3 months ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆71Updated 4 months ago
• InternLM / turbomind
  ☆95Updated 6 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆123Updated 4 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆97Updated 3 months ago
• INT-FlashAttention2024 / INT-FlashAttention
  ☆82Updated 8 months 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.
  ☆44Updated 3 months ago
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆118Updated 4 months ago
• Dao-AILab / grouped-latent-attention
  ☆126Updated 4 months ago
• tile-ai / TileOPs
  ☆51Updated last week
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆230Updated this week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆115Updated last year
• svg-project / flash-kmeans
  Fast and memory-efficient exact kmeans
  ☆48Updated last week
• xdit-project / DiTCacheAnalysis
  An auxiliary project analysis of the characteristics of KV in DiT Attention.
  ☆32Updated 10 months ago
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆105Updated 6 months ago
• triton-lang / kernels
  ☆90Updated 10 months ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆110Updated last year
• ruipeterpan / marconi
  Artifact for "Marconi: Prefix Caching for the Era of Hybrid LLMs" [MLSys '25 Outstanding Paper Award, Honorable Mention]
  ☆37Updated 6 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
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆143Updated 4 months ago
• IST-DASLab / qutlass
  QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning
  ☆99Updated 3 weeks ago
• tilde-research / nsa-impl
  An efficient implementation of the NSA (Native Sparse Attention) kernel
  ☆115Updated 3 months ago
• Infini-AI-Lab / MagicDec
  [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding
  ☆128Updated 10 months ago