FlashAttention

Attention mechanism

Paper: Dao2022flashattention