FlashAttentionはタイリングと再計算を活用してAttention計算を高速化し、メモリ使用量をシーケンス長に対して二次的から線形的な増加に削減する点が特徴です。
Attention実装の効率性を比較したい相談
メモリ使用量の二次から線形への削減