numpy
torch

[dev]
pytest
twine
wheel
