torch>=1.12.0
torchvision>=0.13.0
numpy>=1.21.0
rich>=12.0.0
torchinfo>=1.7.0
scikit-learn>=1.0.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=22.0.0
flake8>=4.0.0
mypy>=0.950
build>=0.8.0
twine>=4.0.0

[examples]
torchvision>=0.13.0
scikit-learn>=1.0.0

[full]
torchinfo>=1.7.0
matplotlib>=3.5.0
seaborn>=0.11.0
tensorboard>=2.8.0
onnx>=1.12.0
safetensors>=0.3.0
sphinx>=5.0.0
sphinx-rtd-theme>=1.0.0
myst-parser>=0.18.0
