#Distributed Training
#Concise Definition
Distributed training is the practice of training a single neural network across many accelerators by partitioning the data, the model, the optimizer state, or the computation schedule, with the devices coordinating through communication collectives. It is required when a model or its training time exceeds what one device can provide.
#Overview & Significance
Frontier models are trained on clusters of thousands of accelerators because neither their memory footprint nor their compute requirement fits on one chip (see TECH-HW-001 AI Hardware & Accelerator Systems and TECH-SCALE-001 Scaling Laws). The engineering problem is not only to split work but to keep devices busy: communication, idle time in pipelines, and hardware failures all reduce the fraction of peak throughput actually achieved. The standard summary metric is model FLOPs utilisation (MFU), the ratio of the model's useful floating-point operations per second to the hardware's theoretical peak.
Distributed training is a principal determinant of who can build frontier systems, because it couples algorithm design, interconnect topology and operations reliability (see ECON-COMP-001 Compute Economics). This article covers the techniques and their published evidence; it does not tabulate current cluster sizes or chip specifications, which are in the Living Reference.
#Historical Development
Early large-scale deep learning used data parallelism across commodity machines. Goyal et al. (2017) showed that ResNet-50 could be trained on ImageNet in one hour on 256 GPUs with minibatches of up to 8,192 images and no loss of accuracy relative to small-batch baselines, using a linear learning-rate scaling rule with warmup.
As language models outgrew single-device memory, three model-partitioning lines developed. Huang et al. (2018) introduced GPipe for pipeline parallelism, splitting layers across devices and splitting each batch into micro-batches. Shoeybi et al. (2019) introduced Megatron-LM's intra-layer (tensor) parallelism for transformers, training an 8.3-billion-parameter model on 512 GPUs. Rajbhandari et al. (2019) introduced ZeRO, which partitions optimizer state, gradients and parameters across data-parallel workers to remove redundancy. Narayanan et al. (2021) combined tensor, pipeline and data parallelism and introduced an interleaved pipeline schedule. Zhao et al. (2023) described PyTorch's fully sharded data-parallel implementation (FSDP), a production system based on the same sharding idea. Jiang et al. (2024) and Grattafiori et al. (2024) documented the operational problems of clusters beyond ten thousand GPUs.
#Technical / Conceptual Explanation
#4.1 Memory budget
With mixed-precision training (Micikevicius et al., 2018) and an Adam-type optimizer, each parameter requires storage not only for weights and gradients but also for optimizer state (fp32 master weights and two moments). The ZeRO analysis (Rajbhandari et al., 2019) counts this as roughly 16 bytes per parameter in total, before activations; a model with parameters therefore needs about bytes for these states. Activations add memory proportional to batch size, sequence length and depth. Activation (gradient) checkpointing recomputes selected activations during the backward pass, trading extra computation for memory; Chen et al. (2016) show memory cost sublinear in network depth with roughly one additional forward pass.
#4.2 Data parallelism
Each of workers holds a full model replica, processes a different shard of the batch, and averages gradients with an all-reduce collective before the optimizer step. For a ring all-reduce over a message of bytes, each worker sends and receives about bytes, nearly independent of . Data parallelism increases the global batch size, and large batches need learning-rate adjustment; Goyal et al. (2017) give the linear scaling rule: when the minibatch is multiplied by , multiply the learning rate by , with warmup. There is a limit beyond which larger batches stop reducing steps proportionally; that limit depends on the task and is a topic of scaling-law research.
#4.3 Sharded data parallelism (ZeRO and FSDP)
ZeRO partitions the optimizer state (stage 1), gradients (stage 2) and parameters (stage 3) across the data-parallel group, gathering parameters just in time for each layer. Rajbhandari et al. (2019) report training models above 100 billion parameters on 400 GPUs at about 15 petaflops, and, without model parallelism, models up to 13 billion parameters; these are 2019 results on the authors' hardware. FSDP implements fully sharded data parallelism in PyTorch; its authors report it matches standard data-parallel throughput while supporting much larger models, with near-linear scaling in TFLOPS in their experiments (Zhao et al., 2023).
#4.4 Tensor (intra-layer) parallelism
Matrix multiplications inside a layer are split across devices. In a transformer MLP block , splitting by columns and by rows lets each device compute independently and require a single all-reduce per block in the forward pass (Shoeybi et al., 2019). The communication volume per layer is high, so tensor parallelism is usually confined to devices connected by the fastest links within a node.
#4.5 Pipeline parallelism
Layers are divided into sequential stages on different devices. A batch is split into micro-batches that flow through the stages; devices are idle during pipeline fill and drain, the bubble, whose fraction in the simple GPipe schedule is (Huang et al., 2018). Interleaved schedules assign several non-contiguous layer chunks to each device, and Narayanan et al. (2021) report that their interleaved schedule improves throughput by more than 10% at comparable memory.
#4.6 Combining the dimensions
Frontier-scale runs combine several parallelisms ("3D" or "4D" parallelism, the latter adding context or sequence parallelism). Narayanan et al. (2021) report a 1-trillion-parameter model on 3,072 GPUs at 502 petaFLOP/s, 52% of theoretical peak per-GPU throughput. Grattafiori et al. (2024) describe 4D parallelism (tensor, context, pipeline, data via FSDP) for their Llama 3 models, and Jiang et al. (2024) report 55.2% MFU for a 175-billion-parameter model on 12,288 GPUs, a 1.34 times improvement over Megatron-LM in their comparison. These are results from 2021–2024 on specific hardware and software and should not be read as current limits.
#4.7 Reliability and stragglers
At the scale of thousands of devices, failures are routine. Grattafiori et al. (2024) report 466 job interruptions during a 54-day pre-training snapshot of Llama 3 training, 419 of them unexpected, with hardware faults (GPU and associated components) the largest category, while maintaining effective training time above 90%. Jiang et al. (2024) describe diagnostic tooling, fault tolerance and straggler mitigation as core components of their system. Frequent checkpointing and automated restart are therefore part of training design.
#Types, Components & Classifications
- Data parallelism: replicate model, split data; all-reduce on gradients.
- Sharded data parallelism: ZeRO stages 1–3 and FSDP; split states, not computation.
- Tensor parallelism: split matrix operations within layers.
- Pipeline parallelism: split by depth; schedules include GPipe-style and interleaved.
- Sequence/context parallelism: split the sequence dimension for long contexts (named in Grattafiori et al., 2024; not further detailed here).
- Expert parallelism: place different experts of a mixture-of-experts model on different devices (see TECH-MOE-001 Mixture-of-Experts Models).
- Synchronous versus asynchronous updates: the large-scale results cited here use synchronous training.
- Memory techniques: mixed precision, activation checkpointing, offloading.
#Applications & Concrete Examples
- Open documentation of production runs: the Llama 3 technical report (Grattafiori et al., 2024) is a primary-source description of parallelism configuration and failure statistics for a large run; the reproduced numbers are limited to what the report states.
- System co-design: MegaScale (Jiang et al., 2024) is a published production system by an industrial team.
- Library support: PyTorch FSDP (Zhao et al., 2023) and the Megatron-LM and ZeRO implementations are the principal open implementations discussed in the cited papers.
- Training runs of closed frontier models are generally undisclosed in cluster size, parallelism layout and utilisation.
#Advantages, Limitations & Unresolved Questions
Advantages. Enables models and training durations impossible on one device, and shortens wall-clock time (Goyal et al., 2017; Narayanan et al., 2021).
Limitations. (1) Communication overhead and pipeline bubbles keep utilisation well below 100% (the published figures above are 30%–55%, depending on setup). (2) Sub-linear returns from large batches. (3) Numerical and hyperparameter fragility at scale; loss spikes and silent hardware errors are operational hazards (Grattafiori et al., 2024; Jiang et al., 2024). (4) Capital and energy requirements concentrate frontier training in few organisations. (5) Reported MFU values from different papers are not directly comparable because model shapes, sequence lengths and accounting conventions differ.
Unresolved. How far synchronous training scales on a single site versus geographically distributed or asynchronous training; how far algorithmic changes (for example communication-reduced optimizers) can lower interconnect requirements. These are active areas; this article cites no claim on either.
#Current State of Knowledge
As of 2026-10-15, the four parallelism families and their trade-offs are established and widely implemented (section 12). Utilisation of roughly 40–55% of peak at ten-thousand-GPU scale is documented in the cited 2021–2024 papers; later improvements and current cluster scales are not tracked here, and the published numbers should be checked against newer technical reports. Hardware specifications, interconnect bandwidths and cluster sizes are in FACT-001 and MODEL-001 and the hardware principles in TECH-HW-001 AI Hardware & Accelerator Systems.
#Future Trajectory & Strategic Outlook
Observed trends. Published systems have moved from hundreds to over ten thousand devices (Shoeybi et al., 2019: 512 GPUs; Jiang et al., 2024: 12,288 GPUs), with reliability engineering becoming a first-class concern.
Near term and beyond. Extrapolating cluster size requires assumptions about power, capital and supply that are outside this article. No forecast is asserted and no FCST- record is attached.
Sources, evidence and revision historyReferences · Evidential Classification Matrix · Revision History
#References & Foundational Literature
- Goyal, P., Dollár, P., Girshick, R., Noordhuis, P., Wesolowski, L., Kyrola, A., Tulloch, A., Jia, Y., & He, K. (2017). Accurate, Large Minibatch SGD: Training ImageNet in 1 Hour. arXiv:1706.02677. [Linear learning-rate scaling rule with warmup; batches up to 8,192 on 256 GPUs.]
- Huang, Y., Cheng, Y., Bapna, A., Firat, O., Chen, M. X., Chen, D., Lee, H., Ngiam, J., Le, Q. V., Wu, Y., & Chen, Z. (2018). GPipe: Efficient Training of Giant Neural Networks using Pipeline Parallelism. NeurIPS 2019. https://arxiv.org/abs/1811.06965. [Pipeline parallelism with micro-batching.]
- Shoeybi, M., Patwary, M., Puri, R., LeGresley, P., Casper, J., & Catanzaro, B. (2019). Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism. arXiv:1909.08053. [Intra-layer tensor parallelism; 8.3B parameters on 512 GPUs.]
- Rajbhandari, S., Rasley, J., Ruwase, O., & He, Y. (2019). ZeRO: Memory Optimizations Toward Training Trillion Parameter Models. SC20 (arXiv:1910.02054). [Partitioning optimizer state, gradients and parameters.]
- Narayanan, D., Shoeybi, M., Casper, J., LeGresley, P., Patwary, M., Korthikanti, V. A., Vainbrand, D., Kashinkunti, P., Bernauer, J., Catanzaro, B., Phanishayee, A., & Zaharia, M. (2021). Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LM. SC21. https://arxiv.org/abs/2104.04473. [Combined parallelism; 1T parameters on 3,072 GPUs at 52% of peak; interleaved pipeline schedule.]
- Micikevicius, P., Narang, S., Alben, J., Diamos, G., Elsen, E., Garcia, D., Ginsburg, B., Houston, M., Kuchaiev, O., Venkatesh, G., & Wu, H. (2018). Mixed Precision Training. ICLR 2018. https://arxiv.org/abs/1710.03740. [Half-precision training with fp32 master weights.]
- Chen, T., Xu, B., Zhang, C., & Guestrin, C. (2016). Training Deep Nets with Sublinear Memory Cost. arXiv:1604.06174. [Activation checkpointing and its memory-compute trade-off.]
- Zhao, Y., Gu, A., Varma, R., Luo, L., Huang, C.-C., Xu, M., Wright, L., Shojanazeri, H., Ott, M., Shleifer, S., Desmaison, A., Balioglu, C., Damania, P., Nguyen, B., Chauhan, G., Hao, Y., Mathews, A., & Li, S. (2023). PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallel. Proceedings of the VLDB Endowment (arXiv:2304.11277). [Production sharded data parallelism in PyTorch.]
- Jiang, Z., Lin, H., Zhong, Y., Huang, Q., Chen, Y., Zhang, Z., Peng, Y., Li, X., Xie, C., Nong, S., Jia, Y., He, S., Chen, H., Bai, Z., Hou, Q., Yan, S., Zhou, D., Sheng, Y., Jiang, Z., Xu, H., Wei, H., Zhang, Z., Nie, P., Zou, L., Zhao, S., Xiang, L., Liu, Z., Li, Z., Jia, X., Ye, J., Jin, X., & Liu, X. (2024). MegaScale: Scaling Large Language Model Training to More Than 10,000 GPUs. NSDI 2024 (arXiv:2402.15627). [55.2% MFU for 175B on 12,288 GPUs; fault tolerance and diagnostics.]
- Grattafiori, A., Dubey, A., Jauhri, A., et al. (Llama Team, AI @ Meta) (2024). The Llama 3 Herd of Models. arXiv:2407.21783 (561 listed authors on the arXiv record). [4D parallelism and interruption statistics from a 54-day pre-training snapshot.]
#Evidential Classification Matrix
| Section / Claim | Evidence Category | Supporting Source | Confidence / Status |
|---|---|---|---|
| All-reduce data parallelism; ring all-reduce communication cost | Established | Goyal et al. (2017); standard collective-communication analysis | High |
| Linear scaling rule works up to batch 8,192 for ResNet-50 | Well Supported (bounded to tested setting) | Goyal et al. (2017) | Moderate to High |
| Tensor and pipeline parallelism designs; pipeline bubble formula | Established (algorithmic) | Shoeybi et al. (2019); Huang et al. (2018) | High |
| ZeRO/FSDP sharding reduces per-device memory | Established (algorithmic); throughput claims Well Supported | Rajbhandari et al. (2019); Zhao et al. (2023) | High / Moderate |
| Utilisation figures (52%, 55.2%) | Well Supported for stated setups; not current limits | Narayanan et al. (2021); Jiang et al. (2024) | Moderate |
| Failure rates at 10k+ GPU scale | Well Supported (two industrial reports) | Grattafiori et al. (2024); Jiang et al. (2024) | Moderate |
| Closed-model training configurations | Undisclosed | n/a | n/a |
#Revision History
- v0.1.0 (2026-10-15): Initial draft. Citations checked against arXiv/Crossref/publisher pages where marked; not independently fact-checked.