DeepSpeed

[BUGFIX] Reset `bucket.elements` after reduction in ZeRO Stage 3

#7418

Merged