Megatron-DeepSpeed

5d65de54 - Merge internal main and github main back into one branch.