GitHub repository: shreyansh26/FlashAttention-PyTorch
Primary language: Python
Description: Implementation of FlashAttention in PyTorch