search

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
8 stars 1 forks source link

automatically `vmap` reconstruction functions if given 3d fields #52

Closed vadmbertr closed 8 months ago

vadmbertr commented 9 months ago

for geostrophy, it should be possible to re-write every thing (including operators) to include 3d fields (but is it really necessary computationally speaking?)

vadmbertr commented 8 months ago

better to leave control of this kind of thinghs to the end-user