DeepSpeed

ce851db2 - Merge branch 'master' into olruwase/zero2_trainable_parameters