Module 13 of 16

AI infrastructure

The infrastructure under models: a training cluster, a data platform, a multi-model inference platform, and the GPU scheduler that shares them.

4 lessons59 min reading

Under every model is a large, expensive, failure-prone machine. This module designs four parts of it. A training cluster sized from the compute a run needs, the network its collectives need, the storage its checkpoints need, and the failure rate that decides how often it stops. A data platform that deduplicates, records lineage, and keeps evaluation data out of training data. An inference platform that serves many models in several regions. And the GPU scheduler that shares all of it, where gang scheduling, preemption, and fragmentation decide how much of the fleet does useful work.

The lessons work in arithmetic: GPU-hours, bytes per parameter, all-reduce volume, mean time between failures, replicas per region. By the end you should be able to size each piece and explain where its capacity goes.

Lessons

  1. 13.aA training clusterHow many GPUs, how much network and storage, and how much reliability engineering does a large training run need?15 min
  2. 13.bA data platformHow do you build training datasets that are deduplicated, free of evaluation data, and traceable from any model back to its sources?15 min
  3. 13.cAn inference platformHow do you serve many models in several regions so that losing a region, a replica, or a model version does not take the product down?14 min
  4. 13.dGPU schedulingHow does a scheduler share a GPU fleet among jobs of very different sizes without deadlock, fragmentation, or idle GPUs?15 min