GitHub repository: raayandhar/SplashAttention
Primary language: Jupyter Notebook
Description: Implementation of Sparse Flash (Splash) Attention in CUDA. FP32, nothing production grade.