torch
numpy

[dev]
pytest
black
isort
