DeepSpeed

0d9fc3ba - Merge branch 'master' into samyamr/fix-for-fragmented-linear-inputs