Speed up Kimi Delta Attention calculations during large language model training or inference.
Drop in as a faster backend for the flash-linear-attention library without changing application code.
Benchmark attention kernel performance on NVIDIA GPUs.
Debug which attention backend is active using built-in logging.
| moonshotai/flashkda | nvidia/cuopt | blinkdl/rwkv-cuda | |
|---|---|---|---|
| Stars | 1,208 | 903 | 232 |
| Language | Cuda | Cuda | Cuda |
| Last pushed | — | — | 2025-12-10 |
| Maintenance | — | — | Quiet |
| Setup difficulty | hard | hard | hard |
| Complexity | 5/5 | 4/5 | 4/5 |
| Audience | researcher | developer | researcher |
Figures from each repo's GitHub metadata at analysis time.
Requires a recent NVIDIA GPU (SM90+), CUDA 12.9+, and compiling from source.
FlashKDA is a specialized piece of software that makes a particular type of AI model calculation run much faster on modern graphics cards, or GPUs, made by NVIDIA. It focuses on something called Kimi Delta Attention, which is a mechanism used inside large language models, the kind of AI systems that power chatbots and text generation tools, to help the model decide which parts of the input text matter most. FlashKDA rewrites that calculation as a highly optimized GPU kernel, meaning a small, tightly tuned piece of code that runs directly on the graphics card hardware, so the same calculation finishes faster and uses resources more efficiently than a general purpose implementation would. The project is built to plug directly into an existing library called flash-linear-attention, which many AI researchers already use to build and run these attention mechanisms. Once installed, FlashKDA is picked up automatically as a faster backend for a function in that library, with no extra code changes needed beyond enabling it, and there is a fallback path available if someone wants to turn it off and use the slower default version instead. Because it works directly with GPU hardware instructions, FlashKDA has demanding requirements: a fairly recent NVIDIA GPU architecture, a recent version of CUDA, NVIDIA's software toolkit for programming GPUs, and a recent version of PyTorch, the machine learning framework it integrates with. Installing it involves cloning the code, pulling in some additional required components, and compiling it directly on a machine with a compatible GPU, which can take some time. The README does not mention a license, so it is unclear under what terms the code can be reused. This project is aimed squarely at AI researchers and engineers who are already building or training large language models and need the underlying math to run as fast as possible on GPU hardware. It assumes deep familiarity with machine learning infrastructure and is not something a beginner or non-technical reader would use directly.
A GPU kernel library that speeds up a specific attention calculation used inside large language models, built for researchers optimizing AI training and inference.
Mainly Cuda. The stack also includes CUDA, PyTorch, CUTLASS.
No license is stated in the README, so reuse and redistribution terms are unclear.
Setup difficulty is rated hard, with roughly 1h+ to a first successful run.
Mainly researcher.
This repo across BitVibe Labs
Don't trust strangers blindly. Verify against the repo.