torch>=1.8.0
torchmetrics
joblib
torchvision
compressai
torchaudio
seaborn
numpy
