Skip to content
English

Switch theme

// ▲ Pretraining, Scaling & Distributed Training

ZeRO / FSDP (Sharded Data Parallelism)

Shards optimizer states, gradients and parameters across GPUs.

ZeRO / FSDP 数据并行 2019 Advanced core

IN PLAIN WORDS

Plain data parallelism replicates everything per GPU. ZeRO's three stages shard states, gradients and parameters, fetching them on demand. PyTorch FSDP is the equivalent.

Used in: DeepSpeed, PyTorch FSDP

TIMELINE · prerequisites → this → what builds on it

2017 Mixed Precisi…2019 ZeRO / FSDP (Shar…

01Original paper

ZeRO: Memory Optimizations Toward Training Trillion Parameter Models

Samyam Rajbhandari, Jeff Rasley, Olatunji Ruwase, et al. · 2019

arXiv ↗ ✓ Verified · arXiv API

02Where it sits

Learn the left column first; the right column builds on this one.

Learn first

ZeRO / FSDP (Sharded Data Parallelism)

Then you can learn

  • Nothing in the atlas builds on it yet

Related:Tensor Parallelism (Megatron-LM)Pipeline Parallelism (GPipe)

Mixed Precision…Tensor Parallel…Pipeline Parall… ZeRO / FSDP (Sharde…

Left: prerequisites · right: builds on this · dashed: related. Full map →

03Learning resources

Official docs, explainers, courses and reference code. All links checked.

04Where the course teaches it

On learning paths:Understanding an LLM End to End

For AI / data

Data updated: AI algorithms & curriculum(UTC+8)