3 ms·FlashAttention in PyTorch2 points by shreyansh26 3y agoshreyansh26 3y agoA simplified implementation of FlashAttention in PyTorch for educational purpose.