flax
jax
jaxopt
optax
scikit-learn
tqdm
typing-extensions

[dev]
coreax[doc,test]
black
isort
pre-commit
pydocstyle
pylint
pyroma

[doc]
coreax[test]
furo
myst-parser
sphinx
sphinx-autodoc-typehints
sphinxcontrib-bibtex

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