Megatron-DeepSpeed

dbb59140 - Merge remote-tracking branch 'origin/main' into thomas/fix_layer_norm