QGFN: Controllable Greediness with Action Values. - Elaine Lau; Stephen Zhewen Lu; Ling Pan; Doina Precup; Emmanuel Bengio (NeurIPS, 2024)

Cell Morphology-Guided Small Molecule Generation with GFlowNets. - Stephen Zhewen Lu; Ziqing Lu; Ehsan Hajiramezanali; Tommaso Biancalani; Yoshua Bengio; Gabriele Scalia; Michal Koziarski (CoRR, 2024)

Structure Language Models for Protein Conformation Generation. - Jiarui Lu; Xiaoyin Chen; Stephen Zhewen Lu; Chence Shi; Hongyu Guo; Yoshua Bengio; Jian Tang 005 (ICLR, 2025)

Aligning Protein Conformation Ensemble Generation with Physical Feedback. - Jiarui Lu; Xiaoyin Chen; Stephen Zhewen Lu; Aurélie C. Lozano; Vijil Chenthamarakshan; Payel Das; Jian Tang 005 (CoRR, 2025)

Measuring Scientific Capabilities of Language Models with a Systems Biology Dry Lab. - Haonan Duan 002; Stephen Zhewen Lu; Caitlin Fiona Harrigan; Nishkrit Desai; Jiarui Lu; Michal Koziarski; Leonardo Cotta; Chris J. Maddison (CoRR, 2025)