← All questions
Training & Fine-tuningMedium· ZeRO· Distributed Training· Communication Efficiency

Explain ZeRO Stage 1 and why it doesn’t cost more bandw...

ZeRO Stage 1 optimizes memory by sharding optimizer states. It does not increase bandwidth usage because it decomposes the standard AllReduce into a ReduceScatter (which delivers per-shard reduced gradients) and an AllGather (which redistributes updated parameters). Because the optimizer update occurs locally on each worker after the ReduceScatter, the total communication remains 2|θ|, identical to a ring-AllReduce.

Code playground (Python)