torch>=2.1
numpy>=1.26
