Alternatives and similar repositories for AttentionEngine:

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

• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆45Updated 5 months ago
• flashinfer-ai / cutlass-viz
  ☆57Updated last week
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆40Updated last week
• DD-DuDa / BitDecoding
  A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.
  ☆34Updated 2 weeks ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆45Updated 9 months ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆69Updated 10 months ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆68Updated this week
• InternLM / turbomind
  ☆84Updated last month
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆84Updated last week
• INT-FlashAttention2024 / INT-FlashAttention
  ☆68Updated 3 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆88Updated last week
• 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.
  ☆36Updated last month
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆38Updated last year
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆100Updated 3 weeks ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆70Updated this week
• mlc-ai / mlc-python
  ☆32Updated last week
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆102Updated this week
• xlite-dev / hgemm-tensorcores-mma
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆74Updated last month
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆122Updated this week
• AlibabaPAI / FLASHNN
  ☆93Updated 7 months ago
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆78Updated 4 months ago
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆67Updated last week
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆91Updated last month
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆108Updated 7 months ago
• thunlp / TritonBench
  TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operators
  ☆36Updated 2 months ago
• stanford-futuredata / stk
  ☆104Updated 8 months ago
• microsoft / chunk-attention
  ☆74Updated 3 weeks ago
• rayleizhu / vllm-ra
  [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts
  ☆39Updated last year
• TiledTensor / TiledCUDA
  We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …
  ☆181Updated 3 months ago
• xdit-project / DiTCacheAnalysis
  An auxiliary project analysis of the characteristics of KV in DiT Attention.
  ☆29Updated 5 months ago