Last released Sep 3, 2026
Differentiable ensemble solvers for differential equations on GPUs, in JAX (record-and-replay adjoint).
Last released Jul 28, 2024
Machine Learning - accelerated Bayesian inference