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