Efficient generative models
Flow matching, diffusion, and discrete diffusion models designed to be cheaper to train and faster at inference time.
PhD Student · Mila · Université de Montréal · Caltech
Jarrid
Rector-Brooks
generative models · sampling · molecular design
I am a PhD student at Mila and the Université de Montréal, supervised by Yoshua Bengio, and I’m currently a visiting researcher in Frances Arnold’s lab at Caltech.
My research is on generative models — efficient algorithms to train and perform inference with them, methods to steer them, and the sampling problems underlying both — across discrete and continuous spaces, oftentimes aimed at designing proteins and other molecular structures. During my PhD I co-founded Dreamfold, and previously studied computer science at the University of Michigan during undergrad.
If you’re interested in working together, feel free to reach out by email.
01 — Focus
What I work on
Steering & control
Guiding pretrained models toward the objectives we care about — controllable, reward-driven generation framed as probabilistic inference (DDPP).
Sampling & inference
Drawing samples from unnormalized, Boltzmann-like densities with neural samplers and amortized inference (iDEM).
Protein & molecular design
Multimodal models that co-design protein sequence and structure for the de novo design of proteins, enzymes, and molecules (DISCO, FoldFlow).
02 — Selected work
Publications
This is probably out of date! For the most up-to-date list, see my Google Scholar.
DISCO
General Multimodal Protein Design Enables DNA-Encoding of Chemistry
PAPL
Planner-Aware Path Learning in Diffusion Language Model Training
OXtal
OXtal: An All-Atom Diffusion Model for Organic Crystal Structure Prediction
Disc2Cont
From Discrete-Time Policies to Continuous-Time Diffusion Samplers: Asymptotic Equivalences and Faster Training
INCO
Inverse-Confidence Sampling for Continuous Diffusion Language Models
DDPP
Steering Masked Discrete Diffusion Models via Discrete Denoising Posterior Prediction
Adversarial Exploration
Adaptive Teachers for Amortized Samplers
BIP-RL
Solving Bayesian Inverse Problems with Diffusion Priors and Off-Policy RL
P2
Path Planning for Masked Diffusion Model Sampling
iDEM
Iterated Denoising Energy Matching for Sampling from Boltzmann Densities
DiffSampler
Improved Off-Policy Training of Diffusion Samplers
FoldFlow-2
Sequence-Augmented SE(3)-Flow Matching for Conditional Protein Backbone Generation
RTB
Amortizing Intractable Inference in Diffusion Models for Vision, Language, and Control
FoldFlow
SE(3)-Stochastic Flow Matching for Protein Backbone Generation
RL Crystals
Learning Conditional Policies for Crystal Design Using Offline Reinforcement Learning
LambdaZero
Generative Active Learning for the Search of Small-Molecule Protein Binders
OT-CFM
Improving and Generalizing Flow-Based Generative Models with Minibatch Optimal Transport
SubTB
Learning GFlowNets from Partial Episodes for Improved Convergence and Stability
MOGFN
Multi-Objective GFlowNets
TS-GFN
Thompson Sampling for Improved Exploration in GFlowNets
FragDiff
Molecular Fragment-Based Diffusion Model for Drug Discovery
RECOVER
RECOVER: Sequential Model Optimization Platform for Combination Drug Repurposing Identifies Novel Synergistic Compounds in vitro
DEUP
DEUP: Direct Epistemic Uncertainty Prediction
ProtGFN
Biological Sequence Design with GFlowNets
Frank-Wolfe
Revisiting Projection-Free Optimization for Strongly Convex Constraint Sets
03 — Beyond the lab
Other stuff
When I’m not building models, you’ll usually find me hiking or biking. I grew up in small-town Michigan and played snare drum in the Michigan Marching Band along the way. Most of all, I’m lucky to share my days with my brilliant partner Alana Valko, our dog Ella, and cat Clementine.
