tqdm
scikit-learn

[:python_version < "3.8"]
torch>=1.13.0

[:python_version >= "3.8"]
torch>=2.4.0
