torch
torchvision
torchaudio
PyYAML
rich
numpy
scipy
