Towards a Python-Based Performance-Portable Finite-Volume Dynamical Core for Numerical Weather Prediction

We present recent progress in the development of a high-performance Python implementation of the non-hydrostatic dynamical core FVM at ECMWF and its member state partners. The FVM formulation is centred about semi-implicit time integration of the fully compressible equations with finite-volume non-oscillatory advection. The numerical schemes can map efficiently to modern supercomputer architectures offering multi-level parallelism and provide competitive time-to-solution. At the same time, the model formulation is considered particularly suitable towards application at convective-scale resolutions. Here, we highlight the new software implementation of FVM in Python leveraging the GT4Py framework and demonstrate its portability and current performance with respect to CPU and GPU based architectures. Furthermore, we discuss aspects of coupling and porting selected ECMWF physical parameterizations to GT4Py.