Megatron-DeepSpeed

189f0547 - Test out the loss from the fp32 weights and optimizer states