torch>=1.8
numpy
matplotlib
tensorboard
scikit-learn
pandas
