Using container images to run PyTorch models in AWS Lambda