torch==2.9.1
matplotlib
