Megatron-DeepSpeed

4a35d50a - Merge branch 'patch-1' of https://github.com/stas00/Megatron-LM into github-pr