jax
equinox>=0.10
tqdm
optax

[dev]
pytest
ruff
sphinx
sphinx-rtd-theme
sphinx-copybutton
nbsphinx
ipython
