DeepSpeed

Fix overlap communication of ZeRO stage 1 and 2

#5606

Merged