DeepSpeed

2fc56841 - Fix (#3527)

Fix (#3527) Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>