DeepSpeed

d6f62217 - Add GPTNeoX AutoTP support (#3778)

Add GPTNeoX AutoTP support (#3778) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Molly Smith <112220543+molly-smith@users.noreply.github.com>