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 ...

[1] Kailai Xu, Bella Shi, Shuyi Yin. 2018. Deep learning for Partial Differential Equations (PDEs). CS230. [2] Maziar Raissi, Paris Perdikaris, and George Em Karniadakis. Physics informed deep ...