DeepSpeed

Fix the universal checkpoint issue for stage3 when there are multiple subgroups.

#7585

Open