torch>=2.0
torchvision>=0.15
numpy

[dev]
pytest
ruff
