torch>=1.9.0
numpy>=1.21.0

[dev]
pytest>=6.0
pytest-cov>=2.0
black>=21.0
flake8>=3.8
isort>=5.0
mypy>=0.812
pre-commit>=2.0

[docs]
sphinx>=4.0
sphinx-rtd-theme>=1.0
sphinx-autodoc-typehints>=1.0
myst-parser>=0.15

[full]
scipy>=1.7.0
scikit-learn>=1.0.0
matplotlib>=3.3.0
seaborn>=0.11.0
pandas>=1.3.0
tqdm>=4.60.0
