torch>=2.0.0

[dev]
pytest
numpy
