torch>=2.1.2
numpy>=1.24.4
