DeepSpeed

9712f10c - Revert "Revert "Merge branch 'master' into staging-1bit-nccl-v2""