
[all]
torch
numpy

[dev]
pytest>=7.0
pytest-cov
torch
numpy

[numpy]
numpy

[torch]
torch
