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