| Metnet: A neural weather model for precipitation forecasting CK Sønderby, L Espeholt, J Heek, M Dehghani, A Oliver, T Salimans, ... arXiv preprint arXiv:2003.12140, 2020 | 288 | 2020 |
| Scaling vision transformers to 22 billion parameters M Dehghani, J Djolonga, B Mustafa, P Padlewski, J Heek, J Gilmer, ... International Conference on Machine Learning, 7480-7512, 2023 | 253 | 2023 |
| Flax: A neural network library and ecosystem for JAX J Heek, A Levskaya, A Oliver, M Ritter, B Rondepierre, A Steiner, ... Version 0.3 3, 14-26, 2020 | 220 | 2020 |
| Efficiently scaling transformer inference R Pope, S Douglas, A Chowdhery, J Devlin, J Bradbury, J Heek, K Xiao, ... Proceedings of Machine Learning and Systems 5, 2023 | 129 | 2023 |
| Deep learning for twelve hour precipitation forecasts L Espeholt, S Agrawal, C Sønderby, M Kumar, J Heek, C Bromberg, ... Nature communications 13 (1), 1-10, 2022 | 104 | 2022 |
| simple diffusion: End-to-end diffusion for high resolution images E Hoogeboom, J Heek, T Salimans International Conference on Machine Learning, 13213-13232, 2023 | 80 | 2023 |
| Skillful twelve hour precipitation forecasts using large context neural networks L Espeholt, S Agrawal, C Sønderby, M Kumar, J Heek, C Bromberg, ... arXiv preprint arXiv:2111.07470, 2021 | 59 | 2021 |
| Bayesian inference for large scale image classification J Heek, N Kalchbrenner arXiv preprint arXiv:1908.03491, 2019 | 45 | 2019 |
| Metnet: A neural weather model for precipitation forecasting. arXiv 2020 CK Sønderby, L Espeholt, J Heek, M Dehghani, A Oliver, T Salimans, ... arXiv preprint arXiv:2003.12140, 0 | 39 | |
| Patch n’pack: Navit, a vision transformer for any aspect ratio and resolution M Dehghani, B Mustafa, J Djolonga, J Heek, M Minderer, M Caron, ... Advances in Neural Information Processing Systems 36, 2024 | 16 | 2024 |
| Skillful twelve hour precipitation forecasts using large context neural networks. arXiv 2021 L Espeholt, S Agrawal, C Sønderby, M Kumar, J Heek, C Bromberg, ... arXiv preprint arXiv:2111.07470, 0 | 10 | |
| Well-calibrated bayesian neural networks J Heek University of Cambridge, 2018 | 5 | 2018 |
| Multistep Consistency Models J Heek, E Hoogeboom, T Salimans arXiv preprint arXiv:2403.06807, 2024 | 1 | 2024 |
| Robustmix: Improving Robustness by Regularizing the Frequency Bias of Deep Nets J Ngnawe, MA Njifon, J Heek, Y Dauphin arXiv preprint arXiv:2304.02847, 2023 | 1 | 2023 |
| Rolling Diffusion Models D Ruhe, J Heek, T Salimans, E Hoogeboom arXiv preprint arXiv:2402.09470, 2024 | | 2024 |
