Implement backend.control_flow.context_overwrite_{while_loop, scan} and create a tutorial for using it

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

Implement backend.control_flow.context_overwrite_{while_loop, scan} and create a tutorial for using it #736

Closed pnkraemer closed 7 months ago

pnkraemer commented 7 months ago

Resolves #674