torch
transformers
accelerate
boto3