Code for the paper

Qu, Y., and X. Shi, 2023: Can a Machine Learning–Enabled Numerical Model Help Extend Effective Forecast Range through Consistently Trained Subgrid-Scale Models?. Artif. Intell. Earth Syst., 2, e220050, https://doi.org/10.1175/AIES-D-22-0050.1.

Barotropic Vorticity Equation in JAX (BVEX)

This project comprises a barotropic vorticity model coded with JAX and machine learning examples.

The BVEX model can be run in a standalone mode and has great speed on a GPU [using run_standalone.py and namelist.py].

A few deep learning (DL) strategies are evaluated to investigate the potentials of DL parameterization of subgrid-scale (SGS) processes.

An example of creating TFDS dataset for DL training is in the highres_forcing_long directory.

An example of creating TFDS dataset for Transfer Learning is in the observation_history directory.

For DL training code -- see DL directory.

For TL training code -- see TL directory;