numpy
matplotlib
scipy
torch>=2.0.0
