torch>=1.10.0

[test]
pytest
pytest-runner
