
[all]
torch
numpy

[numpy]
numpy

[test]
pytest

[torch]
torch
