DeepSpeed

b5a29849 - Merge remote-tracking branch 'refs/remotes/origin/jomayeri/issue-4183' into jomayeri/issue-4183