einops
flax
gradio
huggingface-hub
jax
jaxlib
ml_collections
numpy
orbax-checkpoint
Pillow
psutil
sentencepiece
tensorflow