Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization - Cai, Yuhang; Wu, Jingfeng; Mei, Song; Lindsey, Michael; Bartlett, Peter (Advances in Neural Information Processing Systems, 2024)