requirements.txt

datasets==2.9.0
evaluate==0.4.0
flax==0.6.6
jax==0.4.13
https://storage.googleapis.com/jax-releases/cuda12/jaxlib-0.4.13+cuda12.cudnn89-cp310-cp310-manylinux2014_x86_64.whl
numpy==1.22.4
torch==1.13.1
transformers==4.26.0
streamlit==1.38.0
optax==0.1.4
orbax==0.1.1
sentencepiece==0.1.97