JAX nightlies moving to a new index

[MichaelHudgins]

Hello JAX users!

If you do not use JAX nightlies or install old versions of JAX via a non PyPI index you can stop reading now.

We switching over the JAX nightly pipeline to upload to a new set of simple PEP 503 indexes. The nightly packages will now be published to JAX’s nightly artifact registry. To install these packages, see the JAX installation guide.

For reference the new indexes are listed below. You can add jax or jaxlib to the end of the url to see available versions.

---

A full index including newer JAX nightlies, RCs, releases and a PyPI mirror can found at
https://us-python.pkg.dev/ml-oss-artifacts-published/jax/simple/

Example usage:
python3 -m pip install --pre jax -i https://us-python.pkg.dev/ml-oss-artifacts-published/jax/simple/

---

A nightly only index of JAX can be found at
https://us-python.pkg.dev/ml-oss-artifacts-published/jax-public-nightly-artifacts-registry/simple/

Example usage:
python3 -m pip install --pre jax --extra-index-url https://us-python.pkg.dev/ml-oss-artifacts-published/jax-public-nightly-artifacts-registry/simple/

---

An RC and release index can be found at:
https://us-python.pkg.dev/ml-oss-artifacts-published/jax-public-release-artifacts-registry/simple/

Example usage:
python3 -m pip install jax==0.6.0rc0 --extra-index-url https://us-python.pkg.dev/ml-oss-artifacts-published/jax-public-release-artifacts-registry/simple/