DeepSpeed

9a36258d - Add ds_megatron_gpt model implementation + make explicit megatron is GPT specific