cyclogeostrophy
Computes the inversion of the cyclogeostrophic balance based on a variational formulation approach, using JAX