numpy<2.1.0,>=1.21.0
pandas<3.0.0,>=1.4.0
PyYAML<7.0,>=5.4
matplotlib
scikit_learn<2.0,>=1.0
scipy<2.0,>=1.7
setuptools<70.0,>=49.6.0
torch<3.0,>=1.9.0
tqdm<5.0,>=4.50.0
typing_extensions>=4.0.0
xgboost==2.1.1

[:python_version >= "3.12"]
checkpointer==2.1.0
