FlashAttention in PyTorch github.com 2 points by shreyansh26 3 years ago · 1 comment Reader PiP Save shreyansh26OP 3 years ago A simplified implementation of FlashAttention in PyTorch for educational purpose.