automatically `vmap` reconstruction functions if given 3d fields

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

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

Closed vadmbertr closed 7 months ago

vadmbertr commented 8 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 7 months ago

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