datasets==1.9.0
transformers==4.10.0
flax==0.3.4
optax==0.0.9
tqdm==4.61.1
numpy==1.19.5
tokenizers==0.10.3
sentencepiece==0.1.96
protobuf==3.17.3
tensorboard==2.6.0
torch==1.9.0
tensorflow==2.6.0