issues
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
7
stars
1
forks
source link
v0.1.2
#57
Closed
vadmbertr
closed
6 months ago
vadmbertr
commented
6 months ago
uses
lax.cond
and
lax.while_loop
(in place of
if/else
and
for
statements) in order to
jit
the whole cyclogeostrophy functions
adds DUACS demo notebook
lax.cond
andlax.while_loop
(in place ofif/else
andfor
statements) in order tojit
the whole cyclogeostrophy functions