numpy
torch

[dev]
tox
black
flake8
flake8-black
numpy
torch_testing

[doc]
sphinx==3.3.1
recommonmark
sphinx_rtd_theme
sphinx_markdown_builder
ipython
nbsphinx
sphinxcontrib_katex
pytorch_sphinx_theme
