--find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html
jax[tpu]
keras>=3
einops
keras-hub
safetensors
huggingface_hub