dm_haiku
gradio
huggingface_hub[hf_transfer]
jax==0.4.25
jaxlib
numpy
sentencepiece