diff --git a/WORKSPACE b/WORKSPACE
index 44c02a0785cb..bdbedc74d013 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -50,7 +50,8 @@ new_local_repository(
 #    curl -L https://github.com/openxla/xla/archive/<git hash>.tar.gz | sha256sum
 #    and update the sha256 with the result.
 
-xla_hash = '8d06f3680ad046ea44f8e7159f52c728bb66c069'
+# (based on openxla commit 6e91ff19dad528ab7d2025a9bb46150618a3bc7d)
+xla_hash = 'a820fc0c16ec40728748874dcde92bc7aa5d18daa01e72e9ca2ed4e706e19f3d'
 
 http_archive(
     name = "xla",
diff --git a/setup.py b/setup.py
index 453a73172cf9..c13cec9afa4f 100644
--- a/setup.py
+++ b/setup.py
@@ -65,10 +65,10 @@
 base_dir = os.path.dirname(os.path.abspath(__file__))
 
 USE_NIGHTLY = True  # whether to use nightly or stable libtpu and jax
-_date = '20250113'
+_date = '20250123'
 _libtpu_version = f'0.0.8'
-_jax_version = f'0.4.39'
-_jaxlib_version = f'0.4.39'
+_jax_version = f'0.5.1'
+_jaxlib_version = f'0.5.1'
 _libtpu_wheel_name = f'libtpu-{_libtpu_version}'
 _libtpu_storage_directory = 'libtpu-lts-releases'