torch>=2.1

[dev]
pytest>=7.0
ruff>=0.1
mypy>=1.0
