flash-attention-triton

0.1.1
1.16k

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

    flash-attention-triton download stats