GitHub repository: tspeterkim/flash-attention-minimal
Primary language: Cuda
Created: 2024-03-07T05:38:50Z
Description: Flash Attention in ~100 lines of CUDA (forward pass only)