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.
Context:probdiffeq.control and probdiffeq.adaptive are always used together, yet they currently live in two different modules.
What: This PR moves all content of probdiffeq.control to probdiffeq.adaptive and renames the controller from <old-name> to control_<old-name>. For example, what was previously probdiffeq.control.integral() is now probdiffeq.adaptive.control_integral().
Implications: Two results: (i) the user experience improves because there are fewer modules to import; (ii) the internal code becomes more clear because adaptive and controls were tightly coupled anyway.
This change breaks the current API, so the next version will be v0.4.0. However, we wait with the release because there might be similar PRs coming very soon.
Context:
probdiffeq.controlandprobdiffeq.adaptiveare always used together, yet they currently live in two different modules.What: This PR moves all content of
probdiffeq.controltoprobdiffeq.adaptiveand renames the controller from<old-name>tocontrol_<old-name>. For example, what was previouslyprobdiffeq.control.integral()is nowprobdiffeq.adaptive.control_integral().Implications: Two results: (i) the user experience improves because there are fewer modules to import; (ii) the internal code becomes more clear because adaptive and controls were tightly coupled anyway.
This change breaks the current API, so the next version will be
v0.4.0. However, we wait with the release because there might be similar PRs coming very soon.