DeepSpeed

f26900ec - Modify normalize kernels to save variance in floating-point for backward