Megatron-DeepSpeed

113c6362 - Add temporary assert to finetuning until it can be fixed.