boto3
numpy
torch
PIL