Megatron-style tensor parallelism inside LLMs trains an MNIST classifier to 86.9% accuracy

zeeshanp_ · x · 2026-11-03

zeeshanp continued experimenting inside simulated "QwenGPUs": since MNIST didn't fit in a single message, Claude split a linear classifier across four QwenGPUs with Megatron-style tensor parallelism, one AllReduce per step.

Related event: QwenGPU: Simulating GPU Training Inside an LLM(4 posts)→

Original post →

More from Fun

Fun channel →