DeepSpeed

4e068623 - Several fixes to unblock CI (#3047)

Several fixes to unblock CI (#3047) Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com> Co-authored-by: Jeff Rasley <jerasley@microsoft.com>