numpy
typing
torch

sphinx-copybutton
sphinx-rtd-theme
