DeepSpeed

efec4629 - Revert "pipe engine _aggregate_total_loss: more efficient loss concatenation (#4327)"