Alternatives and similar repositories for tiny-flash-attention

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

• DD-DuDa / Cute-Learning
  Examples of CUDA implementations by Cutlass CuTe
  ☆270Updated 7 months ago
• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆405Updated 3 weeks ago
• SiriusNEO / Triton-Puzzles-Lite
  Puzzles for learning Triton, play it with minimal environment configuration!
  ☆613Updated last month
• Bruce-Lee-LY / cuda_hgemm
  Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…
  ☆520Updated last year
• reed-lau / cute-gemm
  ☆159Updated 2 months ago
• Cjkkkk / CUDA_gemm
  A simple high performance CUDA GEMM implementation.
  ☆426Updated 2 years ago
• flagos-ai / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆287Updated last year
• ifromeast / cuda_learning
  learning how CUDA works
  ☆369Updated 11 months ago
• ColfaxResearch / cutlass-kernels
  ☆258Updated last year
• flagos-ai / FlagGems
  FlagGems is an operator library for large language models implemented in the Triton Language.
  ☆887Updated this week
• wangzyon / NVIDIA_SGEMM_PRACTICE
  Step-by-step optimization of CUDA SGEMM
  ☆424Updated 3 years ago
• ByteDance-Seed / Triton-distributed
  Distributed Compiler based on Triton for Parallel Systems
  ☆1,322Updated last month
• xlite-dev / ffpa-attn
  🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.
  ☆248Updated 2 weeks ago
• ColfaxResearch / cfx-article-src
  ☆173Updated 8 months ago
• Tencent / hpc-ops
  High Performance LLM Inference Operator Library
  ☆603Updated this week
• AyakaGEMM / Hands-on-GEMM
  ☆144Updated last year
• CalebDu / Awesome-Cute
  ☆112Updated 8 months ago
• microsoft / vattention
  Dynamic Memory Management for Serving LLMs without PagedAttention
  ☆457Updated 8 months ago
• Dao-AILab / quack
  A Quirky Assortment of CuTe Kernels
  ☆772Updated last week
• yzhaiustc / Optimizing-SGEMM-on-NVIDIA-Turing-GPUs
  Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.
  ☆403Updated last year
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆319Updated last week
• mit-han-lab / parallel-computing-tutorial
  ☆179Updated 2 years ago
• madsys-dev / deepseekv2-profile
  ☆155Updated 10 months ago
• Yinghan-Li / YHs_Sample
  Yinghan's Code Sample
  ☆364Updated 3 years ago
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆100Updated last year
• sgl-project / SpecForge
  Train speculative decoding models effortlessly and port them smoothly to SGLang serving.
  ☆659Updated this week
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆148Updated 8 months ago
• sail-sg / zero-bubble-pipeline-parallelism
  Zero Bubble Pipeline Parallelism
  ☆449Updated 8 months ago
• AlibabaPAI / FLASHNN
  ☆105Updated last year
• fanshiqing / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆185Updated last month