jax>=0.4
networkx>=3

[dev]
pytest
equinox>=0.11
cottax[docs,ode,solvers,viz]

[docs]
sphinx
sphinx_autodoc_typehints
myst_nb
furo

[ode]
diffrax>=0.7

[solvers]
optimistix
scipy

[viz]
matplotlib
graphviz
cairosvg
