numpy
torch 
torchvision 
torchaudio
