Flash Attention

An optimized attention algorithm that reduces memory usage and speeds up training.