
[gpu]
torch

[test]
pytest>=7
numpy
