Megatron-DeepSpeed

f320d45a - [requirements] check we test agains the correct deepspeed branch