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 | 298 | 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 | 275 | 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 | 229 | 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 | 141 | 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 | 114 | 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 | 89 | 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 | 60 | 2021 |
Bayesian inference for large scale image classification J Heek, N Kalchbrenner arXiv preprint arXiv:1908.03491, 2019 | 46 | 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 | 42 | |
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 | 17 | 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 |