DeepSpeed

ba634fb7 - Merge branch 'master' into loadams/shuffle-true-dataloader