DeepSpeed

b4fcd98f - Inference PP changes for neox (#1899)

Inference PP changes for neox (#1899) Co-authored-by: Reza Yazdani <reyazda@microsoft.com> Co-authored-by: Reza Yazdani <44502768+RezaYazdaniAminabadi@users.noreply.github.com>