OT-Flow: Fast and Accurate Continuous Normalizing Flows via Optimal Transport. - Derek Onken; Samy Wu Fung; Xingjian Li 005; Lars Ruthotto (AAAI, 2021)

A Neural Network Approach Applied to Multi-Agent Optimal Control. - Derek Onken; Levon Nurbekyan; Xingjian Li 005; Samy Wu Fung; Stanley J. Osher; Lars Ruthotto (ECC, 2021)

A Neural Network Approach for High-Dimensional Optimal Control Applied to Multiagent Path Finding. - Derek Onken; Levon Nurbekyan; Xingjian Li 005; Samy Wu Fung; Stanley J. Osher; Lars Ruthotto (IEEE Trans. Control. Syst. Technol., 2023)

A Neural Network Approach for Stochastic Optimal Control. - Xingjian Li 005; Deepanshu Verma; Lars Ruthotto (SIAM J. Sci. Comput., 2024)

OT-Transformer: A Continuous-time Transformer Architecture with Optimal Transport Regularization. - Kelvin Kan; Xingjian Li 005; Stanley J. Osher (CoRR, 2025)

Optimal Control for Transformer Architectures: Enhancing Generalization, Robustness and Efficiency. - Kelvin Kan; Xingjian Li 005; Benjamin J. Zhang; Tuhin Sahai; Stanley J. Osher; Markos A. Katsoulakis (CoRR, 2025)