equinox<0.11.8
flax
jax
jaxopt
jaxtyping
optax
scikit-learn
tqdm
typing-extensions

[doc]
furo
sphinx
sphinx-autodoc-typehints
sphinx-toolbox
sphinxcontrib-bibtex
sphobjinv

[test]
imageio
matplotlib
numpy
opencv-python-headless
pytest-cov
pytest-rerunfailures
scipy
