pnkraemer / probdiffeq

Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and custom information operators. Compatible with the broader JAX scientific computing ecosystem.
https://pnkraemer.github.io/probdiffeq/
MIT License
30 stars 2 forks source link

Rename the input to jnp.reshape to adhere to the new API #740

Closed pnkraemer closed 2 months ago

pnkraemer commented 2 months ago

https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.reshape.html