TLDR AI · ENGINEERING & RESEARCH

PARALLELIZING TRANSFORMER TRAINING

ezyang.github.io39 min read

Summary from the TLDR AI newsletter

This interactive guide explores data parallelism, FSDP, tensor parallelism, pipeline parallelism, and expert parallelism for training transformers. It focuses on how different hardware and communication patterns determine when each strategy becomes bottlenecked.