flash-attention-triton
Cross-platform FlashAttention-2 Triton implementation for Turing+ GPUs with custom configuration mode
What is flash-attention-triton?
Cross-platform FlashAttention-2 Triton implementation for Turing+ GPUs with custom configuration mode