Flash Attention

Flash Attention organiza os cálculos de atenção com menos movimentação de dados.