torch>=2.5.1

[dev]
pytest>=7.0
black
flake8
mypy
