Danyal Rehman

Banting Postdoctoral Researcher @ Mila – Québec AI Institute ·· Ph.D. @ MIT

I'm a Banting Postdoctoral Researcher at Mila – Québec AI Institute, advised by Yoshua Bengio on the development of diffusion/flow-based and autoregressive generative models for AI for Science. I also hold visiting research appointments at both the Broad Institute of MIT & Harvard and AITHYRA.

Previously, I completed my Ph.D. from the Massachusetts Institute of Technology (MIT). During that time, I was awarded research fellowships from the Martin Family Society of Fellows for Sustainability and the Abdul Latif Jameel (J-WAFS) World Water and Food Systems Lab. Prior to that, I completed my undergraduate degree from the University of Toronto with High Honours.

My Ph.D. focused on the development of physics-informed deep learning methods and accelerated partial differential equations (PDEs) solvers for physics-based applications. Examples include neural differential equation models for molecular/ion transport phenomena (under John Lienhard) and self-supervised learning (SSL) methods with Lie point symmetries for PDEs (under Yann LeCun). More recently, my interests have extended to generative modelling for applications to AI for Science.

Danyal Rehman Danyal Rehman
Hiring Interns I’m looking to recruit 1–2 Master’s or Ph.D. research interns interested in generative models for biology at Mila – Québec AI Institute. If you’re interested, please send me your CV and a brief summary of your research experience via email.

Recent Highlights

Feb 2026 RegFlow and FALCON were accepted at ICLR 2026 as a Poster and an Oral (top 1.2% of all submitted papers), respectively.
Jul 2025 RegFlow was awarded Best Paper Award at the ICML Generative AI for Biology Workshop 2025.
Feb 2025 I was awarded NSERC's Banting Postdoctoral Fellowship (C$140,000) to conduct research on AI for Science.
Jul 2024 I joined Mila – Québec AI Institute to work on AI for Science under Yoshua Bengio.
Mar 2024 I joined the Broad Institute of Harvard & MIT to work on generative models for the life sciences under James Collins.
Dec 2023 I successfully defended my Ph.D. from MIT.
Jun 2023 I started an AI/ML Fellowship at Flagship Pioneering.

Selected Publications

Autoregressive Boltzmann Generators.

Danyal Rehman, Charlie B. Tan, Yoshua Bengio, Avishek Joey Bose, Alexander Tong.

Coming soon.

FALCON: Few-step Accurate Likelihoods for Continuous Flows.

Danyal Rehman, Tara Akhound-Sadegh, Artem Gazizov, Yoshua Bengio, Alexander Tong.

Published at ICLR (2026) (Oral). Presented at NeurIPS – Machine Learning for Structural Biology Workshop (2025).

Efficient Regression-based Training of Normalizing Flows for Boltzmann Generators.

Danyal Rehman, Oscar Davis, Jiarui Lu, Jian Tang, Michael Bronstein, Yoshua Bengio, Alexander Tong, Avishek Joey Bose.

Published at ICLR (2026). Received the Best Paper Award at the ICML Generative AI for Biology Workshop (2025).

A Generative Deep Learning Approach to De Novo Antibiotic Design.

Aarti Krishnan, Melis N. Anahtar, Jacqueline A. Valeri, ..., Danyal Rehman, ..., Felix Wong, James J. Collins.

Published in Cell (2025).

Physics-informed Deep Learning for Multi-species Membrane Transport.

Danyal Rehman and John H. Lienhard.

Published in Chemical Engineering Journal (2024). Presented at ICLR – Physics for Machine Learning Workshop (2023).

Self-supervised Learning with Lie Symmetries for Partial Differential Equations.

Grégoire Mialon, Quentin Garrido, Hannah Lawrence, Danyal Rehman, Yann LeCun, Bobak T. Kiani.

Published at NeurIPS (2023). Presented at NeurIPS – AI for Science Workshop (2023).