torch>=1.4
torchvision
numpy
matplotlib
h5py
sphinxcontrib-katex
