Megatron-DeepSpeed

cd20256b - Adding logging of data used per training dataset + 1 bugfix