Training Optimization: Attention Optimizations

Attention optimizations for training: Flash Attention v1, Flash Attention v2, and Flash Attention v3.