torch>=2.3.1
numpy<=2.0.1,>=1.24.4
datasets==3.1.0
