| --find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html | |
| setuptools==69.5.1 | |
| jax[tpu]==0.4.38 | |
| jaxlib==0.4.38 | |
| flax==0.10.2 | |
| orbax-checkpoint==0.5.23 | |
| optax>=0.2.5 | |
| absl-py>=1.4.0 | |
| numpy>=1.26.4,<2.0.0 | |
| ml-dtypes>=0.4.0 | |
| scipy>=1.12.0 | |
| PyYAML>=6.0.1 | |
| matplotlib>=3.8.0 | |
| requests>=2.31.0 | |
| tqdm>=4.66.0 | |
| pillow>=9.5.0 | |
| gcsfs>=2024.1.0 | |
| datasets>=2.19.0 | |
| huggingface-hub>=0.23.0 | |
| transformers>=4.41.2,<4.45.0 | |
| torch==2.3.0 | |
| torchvision==0.18.0 | |
| wandb==0.16.6 | |
| tensorflow>=2.15.0 | |
| einops>=0.7.0 | |
| sacrebleu>=2.4.0 | |
| rouge-score>=0.1.2 |