rich
numpy
accelerate
scikit-learn
torchmetrics
torch>=1.1.0
