pepy.tech

flash-attention-triton

0.1.0
379

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