numpy>=1.19.4
tensorflow
torch
torchvision
