Earlier arXiv versions used different titles; v3 matches the TMLR camera-ready version. Received a Best Paper Award at the [ICML 2024 Workshop on High-dimensional Learning Dynamics (HiLD): The Emergence of Structure and Reasoning](https://sites.google.com/view/hild2024/home?authuser=0).