Jack (Hao) Bai
haob2 AT illinois DOT edu

Hi there! I’m Jack. I’m a rising third-year Ph.D. student at UIUC CS, advised by Prof. Tong Zhang. I work closely with Aviral Kumar @ CMU MLD, Nan Jiang @ UIUC, and Sergey Levine @ BAIR. This summer I work at Microsoft Research with Spencer Whitehead and Jianfeng Gao.
Recently, I focus my research on enhancing the reasoning & planning capability of intelligent agents with foundation models and reinforcement leanring (RL). I am identified as an empirical RL person but still try to make methods principled.
I received my dual undergrad degree from UIUC and Zhejiang University. During those wonderful years, I was lucky enough to work with Yi Ma @ BAIR, Chengxiang Zhai @ UIUC, and Shilin He @ MSR.
A public up-to-date resume can be found here.
News
Jan 23, 2025 | My second paper on building device control agents with RL, Digi-Q, has been accepted to ICLR 2025! Check out the preprint! This work was done when I visited BAIR, advised by Sergey Levine and Aviral Kumar. |
---|---|
Jan 23, 2025 | My first representation learning paper CRATE-LM has been selected as Oral at CPAL 2025! This work was done when I visited BAIR, advised by Prof. Yi Ma. |
Sep 28, 2024 | My two proud works DigiRL (first author) and RL4VLM (second author) have been accepted to NeurIPS 2024! 🎉🎉🎉 |
Latest Posts
Selected Publications
- EMNLP’23