Optimal Control for Transformer Architectures: Enhancing Generalization, Robustness and Efficiency - Kan, Kelvin; Li, Xingjian; Zhang, Benjamin; Sahai, Tuhin; Osher, Stanley; Katsoulakis, Markos (Advances in Neural Information Processing Systems, 2025)