AI Infra Interviews logo

Distributed Training & Parallelism

34 questions
0 of 34 done · 10 unlocked for you
DONEUNLOCKEDLOCKED

Distributed Training Interview Questions

DDP, ZeRO and FSDP, tensor, pipeline, context and expert parallelism, collectives and their cost, MFU, activation checkpointing, elastic and fault-tolerant training, checkpoint economics and the RL post-training stack. Owning the training run at cluster scale.

Learn how AI infrastructure works through explanations, worked examples and diagrams, then practise applying it to interview questions.

Work through the numbers

Change the assumptions in a free calculator, then explain what the result establishes and what still needs measurement.

A starting path through this topic

Start with a mechanism, then use a worked problem to explain a decision. Access labels below describe what is available without an account.

  1. Compare data, tensor and pipeline parallelism. What does each one shard, what does each one communicate, and where does each one live?

    Draw what each parallelism method shards and what it communicates.

    Public answer
  2. Gradient accumulation versus a bigger per-GPU batch: same result or not, and what changes underneath?

    Compare effective batch size with memory use and synchronization.

    Public answer
You have 10 free answers unlocked here.Sign in free for 10 more · 14 are premium.
01–14Foundationsbuild the vocabulary and basic mechanisms0/14 done
15–26Core loopsapply the ideas to worked interview questions0/12 done
27–34Field scenariosreason through constraints and operational failure scenarios0/8 done
PRACTICE TESTStudied this track? Prove it: a timed Distributed Training & Parallelism test.rotating questions · every answer explained · free with an accountStart →

The concepts behind Distributed Training & Parallelism

Review the ideas used in these questions. Each card shows whether the explanation is free, requires sign-in or is part of Premium.

Foundational
Data Parallelism and DDPReplicas combine gradients from different data. Follow a two-worker update, fix unequal-token weighting, and see why queued gradient buckets can leave more than one reduction exposed.
Core
Sign in
ZeRO and FSDPZeRO and FSDP reduce replicated training state by distributing it across workers. The memory saving comes with communication and temporary-buffer costs.
Advanced
🔒 Premium
Tensor ParallelismTensor parallelism splits operations within a layer across GPUs. Derive the column/row matrix split, count training and inference collectives, and choose a group size from memory, latency and actual interconnect performance.
Advanced
🔒 Premium
Pipeline Parallelism and the BubblePipeline parallelism assigns groups of layers to different devices. Microbatches keep stages busy, while the schedule controls idle periods and activation memory.
Advanced
🔒 Premium
Context and Sequence ParallelismContext parallelism distributes a long sequence across devices. Attention then exchanges the information each rank needs instead of keeping the whole sequence locally.
Advanced
🔒 Premium
Expert Parallelism for MoEExpert parallelism places different MoE experts on different devices. Routing tokens between them adds communication and makes uneven expert demand a performance concern.
Foundational
Collective Communication PrimitivesFollow small vectors through all-reduce, reduce-scatter, all-gather and all-to-all. Separate the result each worker needs from the algorithm-specific byte count and measured completion time.
Advanced
🔒 Premium
Ring vs Tree All-ReduceTrace one chunk through a ring, separate byte balance from startup rounds, and compare tree depth without assuming identical bandwidth. Use measured message sizes and topology to choose.
Unlock all 34 answers · ₹2,000 / $25