179 questions
No questions match those filters.
How does ZeRO stage 3 differ from ZeRO stage 1 in memor...
This is one of the questions in the full AI/ML interview bank. Pro unlocks all 1789 questions; Premium includes the same bank plus the highest daily Practice limit.
See plansZeRO stages represent a progression in memory sharding. Stage 1 focuses on sharding the optimizer states across the data-parallel process group. Stage 2 extends this to include gradients, and Stage 3 shards the model parameters themselves.
While Stage 3 provides the most significant memory savings, it introduces a higher communication overhead, requiring roughly 3x the parameter bytes per step compared to 1x in Stage 1. To make this approach practical at scale, it is necessary to overlap the all-gather communication operations with the forward pass computation.