--find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html
jax[tpu]==0.4.14
torch
transformers>=4.40.0
flax==0.7.2
cached-property
requests
yt-dlp>=2023.3.4