DeepSpeed

cd3a7095 - Fix GPT Neo-X multi-gpu inference (#2401)

Fix GPT Neo-X multi-gpu inference (#2401) Co-authored-by: Reza Yazdani <44502768+RezaYazdaniAminabadi@users.noreply.github.com> Co-authored-by: Jeff Rasley <jerasley@microsoft.com>