DeepSpeed

bab49853 - Merge branch 'master' into loadams/fix-torch-linalg-norm