meom-group / jaxparrow

A package for computing the inversion of the cyclogeostrophic balance based on a variational formulation approach, using JAX.
https://jaxparrow.readthedocs.io/
Apache License 2.0
7 stars 1 forks source link

Optax #34

Closed vadmbertr closed 11 months ago