Megatron-DeepSpeed

c44c85b9 - Merge branch 'layer-norm-auto-sync' into ds_ckpt_reshape-with-layer-norm-auto-sync