I work on LLM post-training data and evaluations at Meta. Before joining Meta, I earned my Ph.D. in Computer Science from the University of Illinois Urbana-Champaign, where I was advised by Prof. Jimeng Sun.
My research focused on LLMs and multimodal foundation models, with health data serving as a high-stakes testbed. I co-lead PyHealth, a 1.6K-star open-source toolkit for healthcare AI.
Selected Publications
See the full list on Google Scholar.
-
PyHealth 2.0: A Comprehensive Open-Source Toolkit for Accessible and Reproducible Clinical Deep Learning
ICML 2026 John Wu, Yongda Fan, Zhenbang Wu, Paul Landes, Eric Schrock, Sayeed Sajjad Razin, Arjun Chatterjee, Naveen Baskaran, Joshua Steier, Andrea Fitzpatrick, Bilal Arif, Rian Atri, Jathurshan Pradeepkumar, Siddhartha Laghuvarapu, Junyi Gao, Adam R. Cross, Jimeng Sun [paper] -
Reinforcement Learning for Out-of-Distribution Reasoning in LLMs: An Empirical Study on Diagnosis-Related Group Coding
NeurIPS 2025 (Spotlight) Hanyin Wang, Zhenbang Wu, Gururaj Kolar, Hariprasad Korsapati, Brian Bartlett, Bryan Hull, Jimeng Sun [paper] -
Instruction Tuning Large Language Models to Understand Electronic Health Records
NeurIPS 2024 Datasets and Benchmarks Track (Spotlight) Zhenbang Wu, Anant Dadu, Mike Nalls, Faraz Faghri†, Jimeng Sun†[paper] -
Multimodal Patient Representation Learning with Missing Modalities and Labels
ICLR 2024 Zhenbang Wu, Anant Dadu, Nicholas Tustison, Brian Avants, Mike Nalls, Jimeng Sun, Faraz Faghri [paper] -
MedCLIP: Contrastive Learning from Unpaired Medical Images and Text
EMNLP 2022 Zifeng Wang, Zhenbang Wu, Dinesh Agarwal, Jimeng Sun [paper]
Open-Source Software
-
PyHealth: A Deep Learning Toolkit for Healthcare Applications
[GitHub]
Professional Experience
-
Meta, Burlingame, CA, USA
Software Engineer, Machine Learning
LLM post-training data and evaluations; previously health AI
Jun 2025 - Present
-
Meta, Burlingame, CA, USA
Research Scientist Intern, with Dr. Sujeeth Bharadwaj & Dr. Pei Yin
Foundation model for biosignals
May 2024 - Aug 2024
-
National Institutes of Health & Data Tecnica International, Bethesda, MD, USA
Machine Learning Research Intern, with Dr. Faraz Faghri & Dr. Mike Nalls
Multimodal learning
May 2023 - May 2024
-
IQVIA, Analytics Center of Excellence, Remote, USA
Machine Learning Research Intern, with Dr. Lucas M Glass
Patient record linkage
May 2022 - Aug 2022