jax
jaxopt
scikit-learn
pynndescent

[dev]
black
flake8
isort
mypy
pytest
pytest-cov
typing-extensions

[docs]
sphinxcontrib-autoprogram
sphinxcontrib-napoleon
sphinx-autodocgen
sphinx-github-style>=1.2.2
sphinx-mdinclude
m2r2
nbsphinx
furo
typing-extensions
IPython
