jax
jaxlib
numpy
scipy
sklearn

[dev]
pytest
pytest-xdist
black
isort
flake8

[docs]
recommonmark
sphinx
sphinx-markdown-tables
sphinx-rtd-theme

[quality]
black
isort
flake8

[testing]
pytest
pytest-xdist
