transformers
datasets
pillow
numpy
torch
torchvision
einops
einops-exts
sentencepiece