Variational Learning Finds Flatter Solutions at the Edge of Stability - Ghosh, Avrajit; Cong, Bai; Yokota, Rio; Ravishankar, Saiprasad; Wang, Rongrong; Tao, Molei; Khan, Mohammad Emtiyaz; Möllenhoff, Thomas (Advances in Neural Information Processing Systems, 2025)