NLP Lineage

October 2019 · Scale and data

ZeRO / DeepSpeed

ZeRO: Memory Optimizations Toward Training Trillion Parameter Models

S. Rajbhandari, J. Rasley, O. Ruwase, Y. He

Date
October 2019
What the date marks
arXiv submission
Kind
Infrastructure
Organisation
Microsoft
Access
MIT
Threads
In the guided tour
Chapter 4. Pretrain once, use everywhere
Learning phases
8. Training from scratch

A milestone: 2 later milestones descend directly from it, and its descendants reach 4 of the 10 threads. How levels are decided

Full summary in progress. Until then, this page shows the milestone's record from the research data.

What it changed

Before it

Data-parallel training replicated optimizer state, gradients and weights on every GPU.

The problem

That replication wastes most of the cluster's memory on identical copies.

What it introduced

Partition optimizer state, gradients and parameters across devices instead of duplicating them.

Why it matters

It removed the memory ceiling on data-parallel training and became the open-source default via DeepSpeed.

What modern models inherit

Sharded training state, now assumed in any large run.

What came next

Used for MT-NLG, BLOOM and much of the open training ecosystem.

Before and after

The milestones it descends from, and the milestones that descend from it. Work developed alongside it is not counted. Weak links are left out unless you include them.

Before: where it came from

ZeRO / DeepSpeed has 1 documented ancestors across 1 chapter, reaching back to Adam in 2014. Showing all of them.

ZeRO / DeepSpeed has 1 documented ancestors across 1 chapter, reaching back to Adam in 2014. Showing all of them.

ZeRO / DeepSpeed has 1 recorded ancestors, including indirect ones, across 1 chapter, reaching back to Adam in 2014. Showing all of them.

ZeRO / DeepSpeed has 1 recorded ancestors, including indirect ones, across 1 chapter, reaching back to Adam in 2014. Showing all of them.

  1. 2019ZeRO / DeepSpeedThis milestone.
  2. 2014AdamZeRO / DeepSpeed built directly on thisChapter 2. Words become vectors
  1. 2019ZeRO / DeepSpeedThis milestone.
  2. 2014AdamZeRO / DeepSpeed built directly on thisChapter 2. Words become vectors

After: what it led to

ZeRO / DeepSpeed has 2 documented descendants across 2 chapters, reaching forward to BLOOM in 2022. Showing all of them.

ZeRO / DeepSpeed has 2 documented descendants across 2 chapters, reaching forward to BLOOM in 2022. Showing all of them.

ZeRO / DeepSpeed has 2 recorded descendants, including indirect ones, across 2 chapters, reaching forward to BLOOM in 2022. Showing all of them.

ZeRO / DeepSpeed has 2 recorded descendants, including indirect ones, across 2 chapters, reaching forward to BLOOM in 2022. Showing all of them.

  1. 2019ZeRO / DeepSpeedThis milestone.
  2. 2022MT-NLGbuilt directly on ZeRO / DeepSpeedChapter 5. Scale becomes a science
  3. 2022BLOOMwas influenced by ZeRO / DeepSpeedChapter 8. The weights get out
  1. 2019ZeRO / DeepSpeedThis milestone.
  2. 2022MT-NLGbuilt directly on ZeRO / DeepSpeedChapter 5. Scale becomes a science
  3. 2022BLOOMwas influenced by ZeRO / DeepSpeedChapter 8. The weights get out

Direct links and their sources

Where it came from

Built directly on

Developed alongside

What it led to

Built on directly by

Influenced

Developed alongside