DeepSpeed

Use `deepspeed.comm` instead of `torch.distributed`

#5225

Merged