DeepSpeed

a5517e36 - Merge branch 'master' into cholmes/fix_reduction_utils_amd