Documented URLs are wrong for TPU in setup.py, please see: https://github.com/google/jax/blob/1bcf8d646dfa9f5c87a42280b37bb125c46d447f/setup.py#L86 as an example.
This should reference the URL documented in: https://github.com/google/jax#pip-installation-google-cloud-tpu
What jax/jaxlib version are you using?JAX 0.3.17
Which accelerator(s) are you using?TPU
Additional system infoPython 3.8
NVIDIA GPU infoNo response
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.3