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.
This change enables running probdiffeq.impl-independent tests (e.g. those for probdiffeq.taylor) without selecting an implementation.
This PR also removes the warning for changing an implementation in a script, because (i) there is nothing wrong with changing an implementation and (ii) we frequently do so for testing, benchmarking, etc.
After this PR, it will be easier to consider removing the global state in probdiffeq.impl altogether (should we wish to do so), but this will be a separate concern.
This change enables running
probdiffeq.impl
-independent tests (e.g. those forprobdiffeq.taylor
) without selecting an implementation.This PR also removes the warning for changing an implementation in a script, because (i) there is nothing wrong with changing an implementation and (ii) we frequently do so for testing, benchmarking, etc.
After this PR, it will be easier to consider removing the global state in probdiffeq.impl altogether (should we wish to do so), but this will be a separate concern.
Users (and developers alike) are not affected.