DeepSpeed

d4ddba41 - Merge branch 'master' into olruwase/zero2_trainable_parameters