Megatron-DeepSpeed

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