torch
torchaudio
torchvision
numpy
