numpy
scipy
pandas
torch>=1.4
scikit-learn
numba
pytorch-lightning>=0.8.1
matplotlib
pillow
