GitHub repository: mahdi-shafiei/probdiffeq
Primary language: Python
Description: 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.