tunax 0.1.0

Differential calibrator of the vertical ocean turbulence closures.

Description

This package provides a framework for calibrating the parameters of vertical physics schemes of ocean circulation models using variational optimization. The parameters are calibrated through the minimization of an 'objective function' which compares model predictions with 'Large Eddy Simulations' (LES). Tunax is written in JAX in order to use automatic differentiation for computing the gradient of the objective function with respect to model parameters.

Package organisation

The closures are implemented in the folder tunax/closures/ and they are wrapped by a 'single column model' (SCM) implemented in tunax/model.py. The calibration part is in tunax/fitter.py. The folder notebooks/ contains example of usage of this framework. The folder docs/ is for the documentation configuration which is available [here] (https://tunax.readthedocs.io/en/latest/).

Installation

Stable version

Tunax is pip installable

pip install tunax

Tunax is installed by default with JAX on CPU.