News

About

I'm a Ph.D. student in Applied Mathematics at Purdue University, advised by Prof. Rongjie Lai. I work on scientific machine learning: training neural networks that learn solution operators, so that one model solves a whole family of control and transport problems instead of one instance at a time.

Recently I've developed a self-supervised, in-context operator learner for stochastic mean-field control, and a neural operator that plans collision-free trajectories for swarms of up to 1,000 spacecraft. Both models train on the control objective itself, with no precomputed solutions. I'm also combining diffusion models with classical optimization to recover point configurations from incomplete or noisy data.

Before Purdue, I spent a year in the Ph.D. program at Rensselaer Polytechnic Institute, and earned a B.A. in Mathematics and Physics from McDaniel College, where my undergraduate research was on analysis on the 2-nacci fractal, a variation of the Fibonacci word fractal. I currently serve as Vice President of Purdue's SIAM Student Chapter and as a mathematics graduate student representative.

Research interests

  • Scientific machine learning
  • Operator learning
  • In-context learning
  • Mean-field control
  • Optimal transport
  • Schrödinger bridges
  • Diffusion models
  • Normalizing flows
  • Euclidean distance geometry
Research projects in detail

Publications & Preprints

  1. Self-supervised In-context Operator Learning for Stochastic Mean-Field Control

    Suyi Gao, Mo Zhou, Rongjie Lai

    arXiv preprint arXiv:2608.18282, 2026

  2. Neural Operator Learning for Collision-Aware Trajectory Planning of Spacecraft Swarms

    Sidhdharth D. Sikka, Suyi Gao, Zehui Lu, Rongjie Lai, Shaoshuai Mou

    arXiv preprint arXiv:2608.00320, 2026

Abstracts, BibTeX and talks

Demos

Pivot Cubes

A browser puzzle: reshape a swarm of squares, one pivot at a time, until it matches the target shape.

Play

Education

  • Purdue University Ph.D. in Applied Mathematics · Advisor: Prof. Rongjie Lai 2023 – present
  • Rensselaer Polytechnic Institute Ph.D. program in Applied Mathematics; transferred to Purdue 2022 – 2023
  • McDaniel College B.A. in Mathematics and Physics (double major) 2018 – 2022

Contact

Mail
Department of Mathematics
Purdue University
150 N. University Street
West Lafayette, IN 47907