
🖋 About Me
I am a third-year CS PhD Student at UC Berkeley, Berkeley AI Research Lab (BAIR) and Lawrence Berkeley National Laboratory (LBNL), advised by Aditi Krishnapriyan. I received my B.Sc. with distinction in Data Science from Duke Kunshan University and Duke University. Previously, I worked as a visiting researcher at Meta FAIR, FAIR Chemistry Group, and earlier a research intern at Microsoft Research Asia, Shanghai AI/ML Group.
My research interest mainly falls on AI for Science. Specifically, I am focusing on improving Machine Learning Interatomic Potentials (MLIPs) architectures and applications by designing scalable and efficient models. More broadly, I am interested in combining ideas from mathematics with machine learning, and using machine learning to solve interdisciplinary problems.
🔥 News
- [Sep 2025] Our new AllScAIP model is at the top of the Open Molecules Leaderboard 🎉
- [Jan 2025] Lawrence Berkeley National Laboratory reported EScAIP in the news!
- [Sep 2024] Our new EScAIP model is accepted by NeurIPS 2024 🎉
- [Sep 2024] Joined as a visiting researcher at Meta FAIR!
- [Aug 2023] Starting my PhD at UC Berkeley!
- Show/Hide More
📝 Publications
A recipe for scalable attention-based ML Potentials: unlocking long-range accuracy with all-to-all node attention
Submitted to ICLR 2026
💬 Talks
The Importance of Being Scalable: Improving the Speed and Accuracy of Neural Network Interatomic Potentials Across Chemical Domains
Virtual, SciML Symposium @ Georgia Tech 2024
Kunshan, China, Learning on Graphs Conference Local Meetups 2024Stable Hyperbolic Neural Networks for Graph Generation and Classification
Tokyo, Japan, ICIAM 2023 Mathematics of Geometric Deep Learning MinisymposiumData Continuity Matters: Improving Sequence Modeling with Lipschitz Regularizer
Kigali, Rwanda, ICLR 2023 Spotlight Presentation, Oral 6 Track 5: Applications & Deep Learning
🏆 Awards and Honors
- [2024.5] NERSC GenAI for Science CFP Award (2500 GPU hrs) - U.S. Department of Energy
- [2023.10] Open Catalyst Challenge 3rd Place - NeurIPS 2023 Competitions
- [2023.5] Graduation with Distinction (Top 5%) - Duke Kunshan University
- [2023.5] Graduation with Latin Honors cum laude - Duke Kunshan University
- [2023.5] Zu Chongzhi Math Signature Work Award - Duke Kunshan Zu Chongzhi Math Center
- Show/Hide More
📖 Educations
2023.8 - present PhD Student in Computer Science, University of California, Berkeley
2019.8 - 2023.5 Dual Degree Undergraduate in Data Science, Duke Kunshan University / Duke University
🏢 Work Experiences
2024.9 - 2025.9 Meta FAIR, FAIR Chemistry Group
2022.5 - 2022.9 Microsoft Research, Shanghai AI/ML Group
2023.1 - 2023.6 Microsoft Research, Shanghai AI/ML Group
🏫 Services
Reviewer/PC Member: NeurIPS {2023, 2024, 2025}, ICLR {2024, 2025, 2026}, ICML {2024, 2025}, AISTATS {2025, 2026}
Journal Reviewer: TNNLS
