transformers==4.25.1
einops
unidecode
ftfy
emoji
pillow
jax==0.3.25
flax==0.6.3
orbax==0.0.23
wandb
tqdm
optax
braceexpand
datasets
black
isort