jax[cpu]
flax
optax
huggingface_hub
gradio
numpy