Skip to content
AI LearningAtlas

Module 07: Distributed Systems

Prerequisites

Outcomes

  • Derive collective semantics and first-order communication costs.
  • Match data, tensor, pipeline, and sharded parallelism to constraints.
  • Design observability and recovery for distributed correctness and performance.

Ordered free resources

  1. PyTorch Distributed Overview · Tier A · Documentation / tutorials / code — learn process groups, collectives, and training APIs.
  2. NVIDIA Collective Communications Library Documentation · Tier A · Documentation / reference — inspect GPU collective semantics and topology concerns.
  3. Deep Learning Systems · Tier A · Course / video / labs — connect communication with computation graphs.
  4. Stanford CS336 — Language Modeling from Scratch · Tier A · Course / notes / assignments — place parallel training choices inside a complete language-model system.

Checkpoints

  • Complete Distributed All-Reduce Reasoning.
  • Produce a communication-volume estimate for two parallelism strategies.
  • Write a triage tree for hang, out-of-memory, and divergence failures.

Self-tests

  1. Why is all-reduce not equivalent to gathering every full tensor on one rank?
  2. Which topology details can invalidate a simple bandwidth estimate?
  3. Where can communication overlap with useful computation?

Capstone

Design a multi-GPU plan for a model that does not fit on one device. Include tensor placement, collectives, topology assumptions, memory and communication budgets, failure recovery, and a small-scale validation experiment.

Next: Evaluation

Explore connectionsGraph and backlinks