torch>=2.1

[benchmark]
torchvision>=0.14

[dev]
pytest>=7.0
pytest-cov>=4.0
