numpy
torch

[dev]
sphinx
sphinx-autodoc-typehints
sphinx-rtd-theme
