Tensor and sequence parallelism — explained with pictures

A modern LLM does not fit on a single GPU. A 70B-parameter model in bf16 needs ~140 GB just for the weights, and during training the…