matplotlib
numpy
seaborn
sklearn
sortedcontainers
sympy>=1.4
torch>=1.4.0
torchvision
