jaxparrow - v0.2.3

- Python
Published by vadmbertr about 1 year ago

jaxparrow - v0.2.2

• add cyclogeostrophic_imbalance function
  • ease a posteri evaluation of the loss per grid cell

- Python
Published by vadmbertr over 1 year ago

jaxparrow - v0.2.1

• Better handling of nan/land value #67

- Python
Published by vadmbertr over 1 year ago

jaxparrow - v0.2.0

• fix mask init
• fully jit.able & rev-mode diff.able

- Python
Published by vadmbertr almost 2 years ago

jaxparrow - v0.1.4

• internal implementation changes in order to ease/optimize the jit operations graph generation
• add some missing docstring

- Python
Published by vadmbertr almost 2 years ago

jaxparrow - v0.1.3

• add EKE kinematic

- Python
Published by vadmbertr almost 2 years ago

jaxparrow - v0.1.2

• 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

- Python
Published by vadmbertr almost 2 years ago

jaxparrow - v0.1.1

- Python
Published by vadmbertr about 2 years ago

jaxparrow - v0.1.0

- Python
Published by vadmbertr about 2 years ago

jaxparrow - v0.0.4

Implementation is now in pure jax, allowing the use of vmap on (cyclo)geostrophy functions

- Python
Published by vadmbertr about 2 years ago

jaxparrow - v0.0.3

• improves executable outputs metadata
• fixes some auto documentation (rtd releated) bugs
• updates notebooks figures
• JOSS paper version

- Python
Published by vadmbertr over 2 years ago

jaxparrow - Use optax optimizers

- Python
Published by vadmbertr over 2 years ago

jaxparrow - Initial release

- Python
Published by vadmbertr over 2 years ago