requirements.txt

pillow==8.2.0
aiofiles==0.4.0
numpy
pandas
git+https://github.com/huggingface/transformers.git
datasets >= 1.1.3
jax>=0.2.8
jaxlib>=0.1.59
flax>=0.3.4
optax>=0.0.9
sentencepiece
mtranslate
torch
torchvision