Phase diagram of early training dynamics in deep neural networks: effect of the learning rate, depth, and width - Kalra, Dayal Singh; Barkeshli, Maissam (Advances in Neural Information Processing Systems, 2023)