File size: 232 Bytes
b0ae254 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
-f https://storage.googleapis.com/jax-releases/libtpu_releases.html jax[tpu]==0.4.5 flax==0.6.7 transformers chex datasets optax orbax ftfy tensorboard Jinja2 --extra-index-url https://download.pytorch.org/whl/cpu torch torchvision |