diffrax
equinox
interpax
jax
jaxtyping
lineax
numpy
xarray
