PhD Student · Embodied AI

Jiaqi Bai

PhD Student at MARS Lab, Nanyang Technological University

Building the future of Embodied AI

About me

PhD Researcher at NTU MARS Lab, advised by Prof. Jianfei Yang, developing generative robot trajectory planning and visuomotor policies.

Explore the integration of physical priors for robust trajectory representation with the performance optimization of generative algorithms (e.g., Diffusion, Flow Matching) for real-time robotics.

News

May 2026 New preprint MARS Policy is out, multimodality only when it matters — see the project page / arXiv.
May 2026 New preprint FLASH Policy is out — see the project page / arXiv.
May 2026 New preprint Feedback World Model is out — see the project page / arXiv.
May 2026 New preprint Gaze2Act is out — see the project page / arXiv.
Jan 2026 Our paper RF-MatID was accepted to ICLR 2026 🎉 — see the arXiv.
Aug 2025 Started my Ph.D. in Embodied AI at MARS Lab, NTU, advised by Prof. Jianfei Yang.

Selected publications

^* Equal contribution · ^† Corresponding author

FLASH: Efficient Visuomotor Policy via Sparse Sampling

Jiaqi Bai^*, Jindou Jia^*, Yuxuan Hu, Gen Li, Xiangyu Chen, Tuo An, Kuangji Zuo, Jianfei Yang^†

arXiv preprint arXiv:2605.15492 · 2026

Project arXiv PDF

See all publications →

Get in touch

I'm open to research collaborations and industry opportunities. The fastest way to reach me is by email.

9.jasonbai@gmail.com