rich
numpy>=1.20.0
accelerate>=0.11.0
scikit-learn
torchmetrics
torch>=1.1.0
