Flash Attention

Flash Attention is a way to organize attention calculations so less data has to move through memory.