pardax

A JAX-native solver for initial value problems.

pardax provides a variety of composable time-stepping schemes that work seamlessly with JAX transformations, along with a familiar interface inspired by scipy.integrate.solve_ivp

Installation

pip install pardax

or with uv:

uv add pardax

Next steps

• Getting started
• Tutorial: Heat equation with backward Euler
• Tutorial: Burgers' equation with a pseudo-spectral method
• Extending the solver
• API reference

Read the source code on GitHub.