DeepSpeed

63a89be1 - Revert "Revert "stage3: efficient compute of scaled_global_grad_norm (#5256)" (#5461)"