
[dev]
pytest>=7.0
pytest-cov

[gpu]
torch
numpy
