Update jax dependency to 0.7.1 to align with tt front ends (#8)

Author: jazpurTT

.github/workflows/_build_torch_xla_3.11.yml

@@ -38,7 +38,7 @@ jobs:
           eval "$(pyenv init -)"
           pyenv install 3.11
           pyenv global 3.11
-          ln -sf $HOME/.pyenv/versions/3.11/bin/python3.11 /usr/local/bin/python3.11
+          ln -sf $(pyenv which python3.11) /usr/local/bin/python3.11
 
           # Install essential packages for Python 3.11
           python3.11 -m pip install --upgrade pip

setup.py

@@ -110,14 +110,14 @@
 # 4. After the local build succeeds, create a PR and wait for the CI result. Fix
 #    CI errors as needed until all required checks pass.
 
-USE_NIGHTLY = True  # Whether to use nightly or stable libtpu and JAX.
+USE_NIGHTLY = False  # Whether to use nightly or stable libtpu and JAX.
 
 _libtpu_version = '0.0.21'
 _libtpu_date = '20250813'
 
 _jax_version = '0.7.1'
 _jaxlib_version = '0.7.1'
-_jax_date = '20250813'  # Date for jax and jaxlib.
+_jax_date = '20250617'  # Date for jax and jaxlib.
 
 if USE_NIGHTLY:
   _libtpu_version += f".dev{_libtpu_date}+nightly"