numpy
torch

[docs]
mkdocs
mkdocstrings[python]
mkdocs-jupyter
