torch>=1.10.1

[test]
pytest
