DeepSpeed

42c5381d - Merge remote-tracking branch 'refs/remotes/origin/jomayeri/issue-4083' into jomayeri/issue-4083