xFormersはAttention・正規化・フィードフォワード等の複数のコンポーネントを組み合わせられる構成要素ライブラリであり、FlashAttentionのような特定のAttention実装をバックエンドの一つとして取り込むことができる点が特徴です。
Attention最適化ライブラリを比較したい相談
複数のAttention実装をバックエンドとして選択できる柔軟性