Megatron-DeepSpeed

20c9bfb3 - Merge remote-tracking branch 'origin/main' into log-grad-norm