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