transformers==4.25.1
einops
unidecode
ftfy
emoji
pillow
jax==0.3.25
flax==0.6.3
orbax==0.0.23
wandb

[dev]
tqdm
optax
braceexpand
datasets[streaming]
black[jupyter]
isort
