chex
dm-haiku
matplotlib
numpy
optax
tensorflow

[test]
black
codecov
coverage
flake8
gitchangelog
isort
mkdocs
mypy
pytest
pytest-cov
pytest-mock
scikit-learn

[tool]
jinja2<3.1.0
mkdocs==1.3.0
mkdocs-autorefs==0.4.1
mkdocs-material==8.3.6
mkdocs-material-extensions==1.0.3
mkdocstrings==0.19.0
mkdocstrings-python==0.7.1
pre-commit==2.9.2
pymdown-extensions==9.5
python-semantic-release
twine
