DeepSpeed

7d9c5fb1 - fix megatron regression, add moe unit test, fix moe ckpt comparison