NLP Lineage

17 September 2019 · Scale and data

Megatron-LM

Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism

M. Shoeybi, M. Patwary, R. Puri, P. LeGresley, J. Casper, B. Catanzaro

Date
17 September 2019
What the date marks
arXiv submission
Kind
Infrastructure
Organisation
NVIDIA
Access
custom open
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 3 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

Models were trained by copying the whole model onto each GPU.

The problem

Past a few billion parameters, the model no longer fits on one device at all.

What it introduced

Split individual layers across GPUs with carefully placed communication, reaching 8.3B parameters on 512 GPUs.

Why it matters

The engineering that made the scaling era physically possible. Almost every large model since uses this or its descendants.

What modern models inherit

Tensor parallelism, still part of every frontier training stack.

What came next

ZeRO complemented it; together they trained MT-NLG, PaLM and beyond.

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

Megatron-LM has 24 documented ancestors across 5 chapters, reaching back to Recurrent networks in 1990. Showing its direct parents and the 7 turning points among them.

Megatron-LM has 24 documented ancestors across 5 chapters, reaching back to Recurrent networks in 1990. Showing all of them.

Megatron-LM has 25 recorded ancestors, including indirect ones, across 5 chapters, reaching back to Recurrent networks in 1990. Showing its direct parents and the 7 turning points among them.

Megatron-LM has 25 recorded ancestors, including indirect ones, across 5 chapters, reaching back to Recurrent networks in 1990. Showing all of them.

  1. 2019Megatron-LMThis milestone.
  2. 2019GPT-2 Turning pointMegatron-LM was influenced by thisChapter 4. Pretrain once, use everywhere
  3. 2019WebTextGPT-2 was influenced by thisChapter 4. Pretrain once, use everywhere
  4. 2018BERT Turning pointMegatron-LM was influenced by thisChapter 4. Pretrain once, use everywhere
  5. 2018GPT-1BERT was influenced by this
    GPT-2 built directly on this
    Chapter 4. Pretrain once, use everywhere
  6. 2018ELMoBERT was influenced by thisChapter 4. Pretrain once, use everywhere
  7. 2018ULMFiTBERT was influenced by thisChapter 4. Pretrain once, use everywhere
  8. 2017Mixed precisionMegatron-LM was influenced by thisChapter 9. Making it run anywhere
  9. 2017CoVeELMo was influenced by thisChapter 3. Attention, then only attention
  10. 2017Transformer Turning pointMegatron-LM built directly on thisChapter 3. Attention, then only attention
  11. 2016Layer normTransformer built directly on this
    GPT-2 built directly on this
    Chapter 3. Attention, then only attention
  12. 2015Residual connectionsTransformer was influenced by thisChapter 3. Attention, then only attention
  13. 2015LM pretraining Turning pointBERT was influenced by thisChapter 2. Words become vectors
  14. 2015BPE tokenizationGPT-2 built directly on thisChapter 2. Words become vectors
  15. 2015Luong attentionTransformer was influenced by thisChapter 3. Attention, then only attention
  16. 2014AdamTransformer was influenced by thisChapter 2. Words become vectors
  17. 2014GloVeELMo built directly on thisChapter 2. Words become vectors
  18. 2014Seq2seq Turning pointTransformer built directly on thisChapter 2. Words become vectors
  19. 2014Attention Turning pointTransformer built directly on thisChapter 3. Attention, then only attention
  20. 2014GRU / encoder-decoderAttention built directly on thisChapter 2. Words become vectors
  21. 2013word2vecELMo was influenced by thisChapter 2. Words become vectors
  22. 2011NLP from scratchELMo was influenced by thisChapter 1. Language as probability
  23. 2003Neural LMNLP from scratch was influenced by this
    word2vec was influenced by this
    Chapter 1. Language as probability
  24. 1997LSTM Turning pointSeq2seq built directly on this
    ELMo built directly on this
    Chapter 1. Language as probability
  25. 1990Recurrent networksLSTM was influenced by thisChapter 1. Language as probability
  1. 2019Megatron-LMThis milestone.
  2. 2019GPT-2 Turning pointMegatron-LM was influenced by thisChapter 4. Pretrain once, use everywhere
  3. 2019WebTextGPT-2 was influenced by thisChapter 4. Pretrain once, use everywhere
  4. 2018BERT Turning pointMegatron-LM was influenced by thisChapter 4. Pretrain once, use everywhere
  5. 2018GPT-1BERT was influenced by this
    GPT-2 built directly on this
    Chapter 4. Pretrain once, use everywhere
  6. 2018ELMoBERT was influenced by thisChapter 4. Pretrain once, use everywhere
  7. 2018ULMFiTBERT was influenced by thisChapter 4. Pretrain once, use everywhere
  8. 2017Mixed precisionMegatron-LM was influenced by thisChapter 9. Making it run anywhere
  9. 2017CoVeELMo was influenced by thisChapter 3. Attention, then only attention
  10. 2017Transformer Turning pointMegatron-LM built directly on thisChapter 3. Attention, then only attention
  11. 2017ConvS2STransformer was influenced by this (indirect influence)Chapter 3. Attention, then only attention
  12. 2016Layer normTransformer built directly on this
    GPT-2 built directly on this
    Chapter 3. Attention, then only attention
  13. 2015Residual connectionsTransformer was influenced by thisChapter 3. Attention, then only attention
  14. 2015LM pretraining Turning pointBERT was influenced by thisChapter 2. Words become vectors
  15. 2015BPE tokenizationGPT-2 built directly on thisChapter 2. Words become vectors
  16. 2015Luong attentionTransformer was influenced by thisChapter 3. Attention, then only attention
  17. 2014AdamTransformer was influenced by thisChapter 2. Words become vectors
  18. 2014GloVeELMo built directly on thisChapter 2. Words become vectors
  19. 2014Seq2seq Turning pointTransformer built directly on thisChapter 2. Words become vectors
  20. 2014Attention Turning pointTransformer built directly on thisChapter 3. Attention, then only attention
  21. 2014GRU / encoder-decoderAttention built directly on thisChapter 2. Words become vectors
  22. 2013word2vecELMo was influenced by thisChapter 2. Words become vectors
  23. 2011NLP from scratchELMo was influenced by thisChapter 1. Language as probability
  24. 2003Neural LMNLP from scratch was influenced by this
    word2vec was influenced by this
    Chapter 1. Language as probability
  25. 1997LSTM Turning pointSeq2seq built directly on this
    ELMo built directly on this
    Chapter 1. Language as probability
  26. 1990Recurrent networksLSTM was influenced by thisChapter 1. Language as probability

After: what it led to

Megatron-LM has 5 documented descendants across 3 chapters, reaching forward to QLoRA in 2023. Showing its direct descendants.

Megatron-LM has 5 documented descendants across 3 chapters, reaching forward to QLoRA in 2023. Showing all of them.

Megatron-LM has 6 recorded descendants, including indirect ones, across 3 chapters, reaching forward to QLoRA in 2023. Showing its direct descendants.

Megatron-LM has 6 recorded descendants, including indirect ones, across 3 chapters, reaching forward to QLoRA in 2023. Showing all of them.

  1. 2019Megatron-LMThis milestone.
  2. 2022MT-NLGbuilt directly on Megatron-LMChapter 5. Scale becomes a science
  3. 2022OPT-175Bwas influenced by Megatron-LMChapter 8. The weights get out
  4. 2022LLM.int8()was influenced by OPT-175BChapter 9. Making it run anywhere
  5. 2022GPTQwas influenced by LLM.int8()Chapter 9. Making it run anywhere
  6. 2023QLoRAbuilt directly on LLM.int8()Chapter 9. Making it run anywhere
  1. 2019Megatron-LMThis milestone.
  2. 2022MT-NLGbuilt directly on Megatron-LMChapter 5. Scale becomes a science
  3. 2022OPT-175Bwas influenced by Megatron-LMChapter 8. The weights get out
  4. 2022LLM.int8()was influenced by OPT-175BChapter 9. Making it run anywhere
  5. 2022GPTQwas influenced by LLM.int8()Chapter 9. Making it run anywhere
  6. 2023llama.cppwas influenced by GPTQ (indirect influence)Chapter 9. Making it run anywhere
  7. 2023QLoRAbuilt directly on LLM.int8()Chapter 9. Making it run anywhere

Direct links and their sources

Where it came from

Built directly on

Influenced by

What it led to

Built on directly by

Influenced

Developed alongside