torch>=1.10.0
numpy
matplotlib
