About

I am a postdoctoral researcher in the Department of Computer Science at Princeton University, where I work with Tom Griffiths. I earned my Ph.D. from the Department of Computer Science at the University of Toronto, where my advisor was Richard Zemel.

I work on the intersection of deep learning and probabilistic modeling to build adaptable and reliable machine learning algorithms. My current research interests include:

1. Building learning algorithms that generalize better with limited data.

2. Designing inference algorithms for quantifying the reliability of AI models.

Research keywords: deep learning, metalearning, continual learning, Bayesian inference, uncertainty quantification, nonparametric statistics.

I am on the academic job market for 2024-2025!

News

Jan. 2025: One paper accepted to Nature Human Behavior.
Sep. 2024: One paper accepted at NeurIPS 2024.
Jul. 2024: One paper accepted at TMLR.
Feb. 2024: Invited talk at Stanford University in the Department of Statistics.
Jan. 2024: Two papers accepted at ICLR 2024.
Sep. 2023: Two papers accepted at NeurIPS 2023.

Selected Publications

Thumbnail image for A Metalearned Neural Circuit for Nonparametric Bayesian Inference.

A Metalearned Neural Circuit for Nonparametric Bayesian Inference

Jake C. Snell, Gianluca Bencomo, Thomas L. Griffiths

Neural Information Processing Systems (NeurIPS) 2024.
PDF
Code
poster
Thumbnail image for Implicit Maximum a Posteriori Filtering via Adaptive Optimization.

Implicit Maximum a Posteriori Filtering via Adaptive Optimization

Gianluca M. Bencomo, Jake C. Snell, Thomas L. Griffiths

International Conference on Learning Representations (ICLR) 2024.
PDF
Code
Thumbnail image for Quantile Risk Control: A Flexible Framework for Bounding the Probability of High-Loss Predictions.

Quantile Risk Control: A Flexible Framework for Bounding the Probability of High-Loss Predictions

Jake C. Snell, Thomas P. Zollo, Zhun Deng, Toniann Pitassi, Richard Zemel

International Conference on Learning Representations (ICLR) 2023.
PDF
Code
Thumbnail image for Bayesian Few-Shot Classification with One-vs-Each Pólya-Gamma Augmented Gaussian Processes.

Bayesian Few-Shot Classification with One-vs-Each Pólya-Gamma Augmented Gaussian Processes

Jake C. Snell, Richard Zemel

International Conference on Learning Representations (ICLR) 2021.
PDF
arXiv
Code
Thumbnail image for Prototypical Networks for Few-shot Learning.

Prototypical Networks for Few-shot Learning

Jake C. Snell, Kevin Swersky, Richard Zemel

Neural Information Processing Systems (NeurIPS) 2017.