transformers>=4.21.0
datasets
flax
optax
torch
torchvision
ftfy
tensorboard
modelcards