Universal Transformer validation cost billions, but MoE scaling still favors layer-loop

teortaxesTex · x · 2026-07-21

## Universal Transformer validation cost billions, but the approach fails to scale to billion-parameter MoE models Zitian Gao says the team spent **billions of USD** validating the classic **Universal Transformer** approach, which they call the **“model-loop.”** The main takeaway: - The model-loop **does not scale well to MoE models** with billions of parameters. - In later pretraining stages, the conventional model-loop **underperforms the layer-loop**. - It also needs **more complicated pipeline parallelism** as models get larger, which can slow training. - Because non-loop models have more freedom in parameterization, comparing them purely on theoretical FLOPs is misleading. - The paper therefore compares against a **non-loop baseline matched on actual training FLOPs**, which the authors считают more practical for industrial-scale pretraining. The discussion frames the layer-loop as both **more infrastructure-friendly** and **better performing** in late-stage pretraining.

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

Original post →

More from Research

Research channel →