torch>=2.0.0

[dev]
pytest
pytest-cov
black
isort
