请讲解 DeepSpeed 的原理，重点说明 ZeRO 的三个阶段以及 offload 机制。

回答思路

1. 准确解释 ZeRO 各阶段对模型状态（参数、梯度、优化器状态）的切分方式
2. 能对比 ZeRO 阶段之间的显存节省与通信开销差异
3. 说明 offload 到 CPU/NVMe 的动机与性能权衡