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

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