torch>=2.0
torchmetrics
numpy
matplotlib
tensorboard
scikit-learn
pandas
torchvision
torchaudio
torchtext
