DeepSpeed

a82f6449 - Fix gpt-j inference issue (#3639)

Fix gpt-j inference issue (#3639) * fix gpt-j inference issue for mlp_gemm_func call * bring back the gpt-j inference-test * fix formatting * fix the neox and pythia injection issue