torch
pillow
transformers
timm
einops
openai==0.28