DeepSpeed

5b7413a4 - Fix gpt-Neox rotary embedding implementation (#2782)

Fix gpt-Neox rotary embedding implementation (#2782) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Jeff Rasley <jerasley@microsoft.com>