torch>=2.0.0
numpy>=1.21.0
scikit-learn>=1.0.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
flake8>=6.0.0

[viz]
matplotlib>=3.5.0
