flax
jax
huggingface_hub
optax
numpy
tokenizers
