Megatron-DeepSpeed

5196f8a5 - initial flo count/logging setup (need to fix model parameter count)