einops
flax
gradio
huggingface-hub
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
#jax[cuda12_pip]~=0.4.25
jax[cuda12_pip]==0.4.25
jaxlib
ml_collections
numpy
orbax-checkpoint
Pillow
psutil
sentencepiece
tensorflow-cpu