Learning Rate Transfer in Normalized Transformers
Abstract
The Normalized Transformer, or nGPT (Loshchilov et al., 2025) achieves impressive training speedups and does not require weight decay or learning rate warmup. However, despite having hyperparameters that explicitly scale with model size, we observe that nGPT does not exhibit learning rate transfer across model dimension and token horizon. To rectify this, we combine numerical experiments with a principled use of alignment exponents (Everett et al., 2024) to revisit and modify the P approach to yperparameter transfer (Yang and Hu, 2021). The result is a novel nGPT parameterization we call GPT. Through extensive empirical validation, we find GPT exhibits learning rate transfer across width, depth, and token horizon.
Related papers
See more related submissions when the market opens.