Alternatives and similar repositories for tiny-flash-attention

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

• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆376Updated 11 months ago
• DD-DuDa / Cute-Learning
  Examples of CUDA implementations by Cutlass CuTe
  ☆229Updated 2 months ago
• SiriusNEO / Triton-Puzzles-Lite
  Puzzles for learning Triton, play it with minimal environment configuration!
  ☆504Updated 9 months ago
• reed-lau / cute-gemm
  ☆132Updated 9 months ago
• Cjkkkk / CUDA_gemm
  A simple high performance CUDA GEMM implementation.
  ☆404Updated last year
• ifromeast / cuda_learning
  learning how CUDA works
  ☆317Updated 6 months ago
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆277Updated last year
• ColfaxResearch / cutlass-kernels
  ☆230Updated last year
• 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…
  ☆473Updated last year
• FlagOpen / FlagGems
  FlagGems is an operator library for large language models implemented in the Triton Language.
  ☆668Updated last week
• CalebDu / Awesome-Cute
  ☆103Updated 3 months ago
• AyakaGEMM / Hands-on-GEMM
  ☆138Updated last year
• ColfaxResearch / cfx-article-src
  ☆139Updated 4 months ago
• xlite-dev / ffpa-attn
  🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.
  ☆212Updated last month
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆95Updated 8 months ago
• wangzyon / NVIDIA_SGEMM_PRACTICE
  Step-by-step optimization of CUDA SGEMM
  ☆373Updated 3 years ago
• yzhaiustc / Optimizing-SGEMM-on-NVIDIA-Turing-GPUs
  Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.
  ☆379Updated 8 months ago
• microsoft / vattention
  Dynamic Memory Management for Serving LLMs without PagedAttention
  ☆414Updated 3 months ago
• ByteDance-Seed / Triton-distributed
  Distributed Compiler based on Triton for Parallel Systems
  ☆1,107Updated this week
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆323Updated 5 months ago
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆165Updated this week
• mit-han-lab / parallel-computing-tutorial
  ☆172Updated 2 years ago
• Yinghan-Li / YHs_Sample
  Yinghan's Code Sample
  ☆345Updated 3 years ago
• AlibabaPAI / FLASHNN
  ☆98Updated last year
• madsys-dev / deepseekv2-profile
  ☆146Updated 6 months ago
• MARD1NO / CUDA-PPT
  ☆108Updated 5 months ago
• nicolaswilde / cuda-sgemm
  ☆69Updated 8 months ago
• sgl-project / SpecForge
  Train speculative decoding models effortlessly and port them smoothly to SGLang serving.
  ☆374Updated this week
• Bruce-Lee-LY / flash_attention_inference
  Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.
  ☆40Updated 6 months ago
• galeselee / Awesome_LLM_System-PaperList
  Since the emergence of chatGPT in 2022, the acceleration of Large Language Model has become increasingly important. Here is a list of pap…
  ☆268Updated 6 months ago