torch
torchvision
numpy
mkdocs
mkdocs-terminal
mkdocstrings[python]
