DeepSpeed

8ea1f5f1 - Merge branch 'master' into samyamr/fix-for-fragmented-linear-inputs