Megatron-DeepSpeed

5b368846 - Merge remote-tracking branch 'origin/main' into olruwase/sync_layer_norms