DeepSpeed

Refactor `gptj_residual_add` kernels for better readability

#2358

Merged