PyBamm embedded GP for parameter estimation
src is current working version, utilizing IdakluJax solver, automatic differentiation
src-non-jax is the previous version with the base Idaklu solver implemented, and sensitivities used for grad calculations