torch
numpy

[test]
pytest
pytest-xdist
