Bayesian Inference for Large Scale Image Classification. - Jonathan Heek; Nal Kalchbrenner (CoRR, 2019)

MetNet: A Neural Weather Model for Precipitation Forecasting. - Casper Kaae Sønderby; Lasse Espeholt; Jonathan Heek; Mostafa Dehghani 001; Avital Oliver; Tim Salimans; Shreya Agrawal; Jason Hickey; Nal Kalchbrenner (CoRR, 2020)

Skillful Twelve Hour Precipitation Forecasts using Large Context Neural Networks. - Lasse Espeholt; Shreya Agrawal; Casper Kaae Sønderby; Manoj Kumar; Jonathan Heek; Carla Bromberg; Cenk Gazen; Jason Hickey; Aaron Bell; Nal Kalchbrenner (CoRR, 2021)

Efficiently Scaling Transformer Inference. - Reiner Pope; Sholto Douglas; Aakanksha Chowdhery; Jacob Devlin; James Bradbury; Anselm Levskaya; Jonathan Heek; Kefan Xiao; Shivani Agrawal; Jeff Dean (CoRR, 2022)

Scaling Vision Transformers to 22 Billion Parameters. - Mostafa Dehghani 001; Josip Djolonga; Basil Mustafa; Piotr Padlewski; Jonathan Heek; Justin Gilmer; Andreas Peter Steiner; Mathilde Caron; Robert Geirhos; Ibrahim Alabdulmohsin; Rodolphe Jenatton; Lucas Beyer; et al. (ICML, 2023)

simple diffusion: End-to-end diffusion for high resolution images. - Emiel Hoogeboom; Jonathan Heek; Tim Salimans (ICML, 2023)

Efficiently Scaling Transformer Inference. - Reiner Pope; Sholto Douglas; Aakanksha Chowdhery; Jacob Devlin; James Bradbury; Jonathan Heek; Kefan Xiao; Shivani Agrawal; Jeff Dean (MLSys, 2023)

Patch n' Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution. - Mostafa Dehghani 001; Basil Mustafa; Josip Djolonga; Jonathan Heek; Matthias Minderer; Mathilde Caron; Andreas Steiner 001; Joan Puigcerver; Robert Geirhos; Ibrahim M. Alabdulmohsin; Avital Oliver; Piotr Padlewski; et al. (NeurIPS, 2023)

Robustmix: Improving Robustness by Regularizing the Frequency Bias of Deep Nets. - Jonas Ngnawé; Marianne Abemgnigni Njifon; Jonathan Heek; Yann N. Dauphin (CoRR, 2023)

Rolling Diffusion Models. - David Ruhe; Jonathan Heek; Tim Salimans; Emiel Hoogeboom (ICML, 2024)

Multistep Distillation of Diffusion Models via Moment Matching. - Tim Salimans; Thomas Mensink; Jonathan Heek; Emiel Hoogeboom (NeurIPS, 2024)

Multistep Consistency Models. - Jonathan Heek; Emiel Hoogeboom; Tim Salimans (CoRR, 2024)

Simpler Diffusion (SiD2): 1.5 FID on ImageNet512 with pixel-space diffusion. - Emiel Hoogeboom; Thomas Mensink; Jonathan Heek; Kay Lamerigts; Ruiqi Gao; Tim Salimans (CoRR, 2024)

Model Integrity when Unlearning with T2I Diffusion Models. - Andrea Schioppa; Emiel Hoogeboom; Jonathan Heek (CoRR, 2024)

Simpler Diffusion: 1.5 FID on ImageNet512 with Pixel-space Diffusion. - Emiel Hoogeboom; Thomas Mensink; Jonathan Heek; Kay Lamerigts; Ruiqi Gao; Tim Salimans (CVPR, 2025)