cyclogeostrophy
A package for computing cyclogeostrophic currents with a minimisation-based approach, using JAX.