numpy
torch
tensorflow
torchvision
torchaudio
