numpy
prefetch-generator

[dev]
pytest
pytest-xdist
pre-commit
torch
