numpy
cppyy
torch>=1.8

[cuda]
cupy

[test]
pytest
