Zuobai Zhang

PhD student, Mila - Université de Montréal

Joined

May 2021

Names

Zuobai Zhang (Preferred)

Emails

****@umontreal.ca (Confirmed)

****@fudan.edu.cn (Confirmed)

****@mila.quebec (Confirmed)

Personal Links

Career & Education History

PhD student

Mila - Université de Montréal (umontreal.ca)

2021 – 2026

Intern

NVIDIA (nvidia.com)

2024 – 2024

Undergrad student

Fudan University (fudan.edu.cn)

2017 – 2021

Advisors, Relations & Conflicts

No relations added

Expertise

Graph Representation Learning

2021 – Present

Drug Discovery

2021 – Present

Publications

Scaling Atomistic Protein Binder Design with Generative Pretraining and Test-Time Compute
Kieran Didi, Zuobai Zhang, Guoqing Zhou, Danny Reidenbach, Zhonglin Cao, Sooyoung Cha, Tomas Geffner, Christian Dallago, Jian Tang, Michael M. Bronstein, Martin Steinegger, Emine Kucukbenli, Arash Vahdat, Karsten Kreis
- ICLR 2026 Oral
- Readers: Everyone
Towards All-Atom Foundation Models for Biomolecular Binding Affinity Prediction
Liang Shi, Zuobai Zhang, Huiyu Cai, Santiago Miret, Zhi Yang, Jian Tang
- ICLR 2026 Poster
- Readers: Everyone
Fast Proteome-Scale Protein Interaction Retrieval via Residue-Level Factorization
Jianan Zhao, Zhihao Zhan, Narendra Chaudhary, Xinyu Yuan, Zuobai Zhang, Qian Cong, Jian Zhou, Sanchit Misra, Jian Tang
- ICLR 2026 Poster
- Readers: Everyone
La-Proteina: Atomistic Protein Generation via Partially Latent Flow Matching
Tomas Geffner, Kieran Didi, Zhonglin Cao, Danny Reidenbach, Zuobai Zhang, Christian Dallago, Emine Kucukbenli, Karsten Kreis, Arash Vahdat
- ICLR 2026 Poster
- Readers: Everyone
Consistent Synthetic Sequences Unlock Structural Diversity in Fully Atomistic De Novo Protein Design
Danny Reidenbach, Zhonglin Cao, Zuobai Zhang, Kieran Didi, Tomas Geffner, Guoqing Zhou, Jian Tang, Christian Dallago, Arash Vahdat, Emine Kucukbenli, Karsten Kreis
- NeurIPS2025-AI4Science Poster
- Readers: Everyone
Proteina: Scaling Flow-based Protein Structure Generative Models
Tomas Geffner, Kieran Didi, Zuobai Zhang, Danny Reidenbach, Zhonglin Cao, Jason Yim, Mario Geiger, Christian Dallago, Emine Kucukbenli, Arash Vahdat, Karsten Kreis
- ICLR 2025 Oral
- Readers: Everyone
Cell ontology guided transcriptome foundation model
Xinyu Yuan, Zhihao Zhan, Zuobai Zhang, Manqi Zhou, Jianan Zhao, Boyu Han, Yue Li, Jian Tang
- AIM-FM Workshop @ NeurIPS'24 Poster
- Readers: Everyone
Cell ontology guided transcriptome foundation model
Xinyu Yuan, Zhihao Zhan, Zuobai Zhang, Manqi Zhou, Jianan Zhao, Boyu Han, Yue Li, Jian Tang
- AIDrugX Poster
- Readers: Everyone
Cell ontology guided transcriptome foundation model
Xinyu Yuan, Zhihao Zhan, Zuobai Zhang, Manqi Zhou, Jianan Zhao, Boyu Han, Yue Li, Jian Tang
- Neurips 2024 Workshop FM4Science Poster
- Readers: Everyone
Augmenting Evolutionary Models with Structure-based Retrieval
Yining Huang, Zuobai Zhang, Jian Tang, Debora Susan Marks, Pascal Notin
- ML4LMS Poster
- Readers: Everyone

View all 31 publications

Co-Authors

View all 62 co-authors