A $100M compute study says layer-loop beats model-loop for large MoE pretraining

teortaxesTex · x · 2026-07-21

The first author says the team spent hundreds of millions of dollars in compute validating the classical Universal Transformer, or “model-loop,” setup. They found that model-loop does not scale well to billion-parameter MoE models. In later-stage pretraining, their “layer-loop” approach outperformed model-loop, stayed ahead through the end of training, and was friendlier to infrastructure and pipeline parallelism. They also argue that, at equal theoretical FLOPs, the looped approach can beat non-loop baselines, and that model-loop places parameters and gradients farther apart, which may slow training as scale grows.

Related event: Loopie Cyclic Transformer Beats 30B Baseline with Equal Compute(6 posts)→

Original post →

More from Infra

Infra channel →