Megatron-DeepSpeed

b9ac381f - formatting fix, reserving bug for somewhere else, adding flo-logging to TB groups