DeepSpeed

4fb83517 - Merge branch 'master' into olruwase/zero2_trainable_parameters