torch
transformers
pillow
einops