
[dev]
pytest

[torch]
torch>=1.10
