DeepSpeed

27bd006c - Merge branch 'master' into samyamr/fix-for-fragmented-linear-inputs