huggingface_hub
numpy
pyyaml
scipy
tqdm

[dev]
pre-commit
pytest
pytest-cov
mypy
flake8
pygount
types-setuptools
types-tqdm
types-toml
types-PyYAML
radon
mkdocs
mkdocs-material
mkdocs-autorefs
mkdocstrings
mkdocs-get-deps
mkdocs-material-extensions
mkdocstrings-python

[torch]
torch>=1.0.0
