DeepSpeed

1efe252d - Merge branch 'master' into olruwase/zero2_trainable_parameters