torch
transformers[torch]
pillow