pepy.tech

flash-attention-triton

0.1.1
1.47k

Cross-platform FlashAttention-2 Triton implementation for Turing+ architectures