jax>=0.6.1
optax>=0.2
equinox>=0.11
optimistix>=0.1.0
lineax>=0.0.7
diffrax>=0.7
paramax>=0.0.5

[docs]
matplotlib>=3.8
jupytext
jupyter_book<2
sphinx_autodoc_typehints
sphinx_math_dollar

[gpu]
jax[cuda12]>=0.6.1

[test]
matplotlib>=3.8
pytest>=8.0
scipy
pre-commit
