Jargonautica
Training and Post-training

Data, Tensor and Pipeline Parallelism

Also called: Model Parallelism, Tensor Parallelism, Pipeline Parallelism

The three ways of splitting a training job across many chips. Data parallelism gives each chip a copy of the model and different examples. Tensor parallelism splits individual layers across chips. Pipeline parallelism gives each chip different layers. Large runs use all three at once.

Technical view

Data parallelism requires gradient synchronisation each step and scales until communication dominates. Tensor parallelism splits matrix multiplications within a layer, demanding very high bandwidth so it is normally kept within a node. Pipeline parallelism partitions by depth and introduces bubble overhead requiring micro-batching to fill. Combining all three is called 3D parallelism, and choosing the configuration is a co-design problem between model shape and network topology.


Part of AI Terminology Index, section: Training and Post-training.

How it connects

Data, Tensor and Pipeline Parallelism links to 5 other terms. The faint lines are links between those terms. See it in the full graph.

TrainingTrainingComputeComputeData Centre and InterconnectData Centre and Int…Sharding and ZeROSharding and ZeROGradient AccumulationGradient Accumulati…Data, Tensor and Pipeline Parallelism