Visualization of Data Parallelism for LLM Training: From Naive Data Parallelism to ZeRO-3

Training large models is usually introduced as a compute problem: more parameters require more FLOPs, so we spread the work across many…