torch>=2.0
torchvision>=0.15
scikit-learn>=1.3

[bench]
matplotlib>=3.8
numpy>=1.26
torchaudio<2.11,>=2.0

[dev]
build>=1.2
pytest>=8.0
ruff>=0.4
twine>=5.0

[docs]
matplotlib>=3.8
