probabilistic-numerics
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.
Probabilistic Numerics in Python.
Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"