DeepSpeed

a43e0fb5 - Merge branch 'master' into olruwase/zero_optional_reduce_scatter