prototorch>=0.7.3
pytorch_lightning>=1.6.0
torchmetrics
protobuf<3.20.0

[all]
jsonargparse
bumpversion
pre-commit
recommonmark
sphinx
nbsphinx
ipykernel
sphinx_rtd_theme
sphinxcontrib-katex
sphinxcontrib-bibtex
matplotlib
scikit-learn
codecov
pytest

[dev]
bumpversion
pre-commit

[examples]
matplotlib
scikit-learn

[tests]
codecov
pytest
