“Flash Attention has been integrated as an optional feature into NVIDIA's Megatron-LM library.”