ProductAI Software / Framework

FlashAttention-2

Optimized GPU attention kernel for transformer models, the second generation of FlashAttention.

Mentioned in