Megatron-DeepSpeed

dedf15c6 - Merge remote-tracking branch 'origin/main' into layer-norm-auto-sync