torch==2.7.1
numpy==1.26.4