Update the jax.config usage to adhere to the new API and fix the doc-build

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

Update the jax.config usage to adhere to the new API and fix the doc-build #741

Closed pnkraemer closed 2 months ago

pnkraemer commented 2 months ago

and set the JUPYTER_PLATFORM_DIRS variable to 1, because a related keeps coming up and breaks the doc-build.