[torchax] Update installation guide in readme (#8664)

Co-authored-by: Siyuan Liu <[email protected]>

Author: lsy323

torchax/README.md

@@ -26,7 +26,7 @@ The following instructions assume you are in the `torchax` directory:
 ```
 Fork the repository
 $ git clone https://github.com/<github_username>/xla.git
-$ cd xla/experimental/torchax
+$ cd xla/torchax
 ```
 
 
@@ -55,17 +55,11 @@ Note: `dev-requirements.txt` will install the CPU-only version of PyTorch.
 
 #### 1.1 Install this package
 
-If you want to install torchax without the jax dependency and use the jax dependency from torch_xla:
-```bash
-pip install torch_xla[pallas] -f https://storage.googleapis.com/jax-releases/jax_nightly_releases.html -f https://storage.googleapis.com/jax-releases/jaxlib_nightly_releases.html
-pip install -e .
-```
-
-Otherwise, install `torchax` from source for your platform:
+Install `torchax` from source for your platform:
 ```bash
 pip install -e .[cpu]
 pip install -e .[cuda]
-pip install -e .[tpu] -f https://storage.googleapis.com/libtpu-releases/index.html
+pip install -e .[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
 ```
 
 #### 1.2 (optional) verify installation by running tests