Implement backend.control_flow.scan

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.scan #685

Closed pnkraemer closed 9 months ago

pnkraemer commented 9 months ago

En route to a cleaner import/backend management in probdiffeq.