coloredlogs>=15.0.1
numpy>=1.24
scikit-learn>=1.3.2
tensorboard>=2.0
torch>=2.1.1
