numpy
jax
jaxlib
flax
pandas
pathos
tqdm

[dev]
flake8
flake8-print
flake8-bugbear
mypy
scikit-learn
pytest
pytest-xdist
pytest-cov
pylint
black
reorder-python-imports
autoflake
pre-commit
pydocstring-coverage
bumpversion
ipython
jupyter
gprof2dot
bandit
check-python-versions
line-profiler

[docs]
sphinx
sphinxcontrib-napoleon
sphinx_rtd_theme
