JAX Solver
============

.. autoclass:: pybamm.JaxSolver
  :members:

.. autofunction:: pybamm.jax_bdf_integrate