Megatron-DeepSpeed

a025c8a1 - dynamically use as many 3d dimensions as possible (#83)