Flash Attention

Flash Attention organise les calculs d’attention en réduisant les déplacements de données.