torch>=2.1.0
numpy