DeepSpeed

65d6266a - Revert "stage3: efficient compute of scaled_global_grad_norm (#5256)"