Megatron-DeepSpeed

efeb55a1 - Merge remote-tracking branch 'official_repo/main' into thomas/mtf_train_script