flax
jax
huggingface_hub
optax
numpy
tiktoken
