Megatron-DeepSpeed

3c5e4914 - Merge remote-tracking branch 'origin/main' into thomas/test_different_layer_norm