torch==1.8
numpy
matplotlib
toml