torch>=2.8.0
matplotlib
numpy
