Geoscientific Model Development
Geoscientific Model Development
GMD
Articles | Volume 18, issue 2
Articles | Volume 18, issue 2
https://doi.org/10.5194/gmd-18-529-2025
© Author(s) 2025. This work is distributed under
the Creative Commons Attribution 4.0 License.
https://doi.org/10.5194/gmd-18-529-2025
© Author(s) 2025. This work is distributed under
the Creative Commons Attribution 4.0 License.
Articles | Volume 18, issue 2
Development and technical paper
 | 
30 Jan 2025
Development and technical paper |  | 30 Jan 2025

Exploring a high-level programming model for the NWP domain using ECMWF microphysics schemes

Stefano Ubbiali, Christian Kühnlein, Christoph Schär, Linda Schlemmer, Thomas C. Schulthess, Michael Staneker, and Heini Wernli

Data sets

Data and scripts for the manuscript "Exploring a high-level programming model for the NWP domain using ECMWF microphysics schemes" (v0.1.0) Stefano Ubbiali et al. https://doi.org/10.5281/zenodo.11155354

Model code and software

ifs-physics-common: v0.1.0 Stefano Ubbiali et al. https://doi.org/10.5281/zenodo.11153742

gt4py-dwarf-p-cloudsc: v0.1.0 Stefano Ubbiali et al. https://doi.org/10.5281/zenodo.11155002

gt4py-dwarf-p-cloudsc2-tl-ad: v0.3.0 Stefano Ubbiali et al. https://doi.org/10.5281/zenodo.13239373

Download
Short summary
We explore a high-level programming model for porting numerical weather prediction (NWP) model codes to graphics processing units (GPUs). We present a Python rewrite with the domain-specific library GT4Py (GridTools for Python) of two renowned cloud microphysics schemes and the associated tangent-linear and adjoint algorithms. We find excellent portability, competitive GPU performance, robust execution on diverse computing architectures, and enhanced code maintainability and user productivity.
Read more