torch>=2.0
numpy>=1.24
scikit-learn>=1.3
tqdm>=4.65