
[dev]
pytest>=7.0.0

[pytorch]
torch>=1.9.0
