Tensor Programs IV revisited: muP enables feature learning in infinite-width networks

burny_tech · x · 2026-11-10

Sharing the foundational muP paper (Greg Yang, Edward J. Hu, ICML 2021) alongside a discussion of whether theory can precede empirics. The paper shows standard and NTK parametrizations admit no feature learning in the infinite-width limit, proposes simple parametrization changes that do, derives explicit limit formulas via the Tensor Programs technique, and demonstrates on Word2Vec and MAML/Omniglot that these limits beat NTK baselines and finite-width networks — the basis for later hyperparameter transfer techniques.

Related event: From muP to Probabilistic Oracles: Theory-First AI Design Revisited(2 posts)→

Original post →

More from AGI Musings

AGI Musings channel →