/jaxparrow

A package for computing the inversion of the cyclogeostrophic balance based on a variational formulation approach, using JAX.

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers