huggingface_hub
numpy
jax
jaxlib
flax
optax
sentencepiece
