diff --git a/WORKSPACE b/WORKSPACE
index 8ca713361af..c4da2027106 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -50,7 +50,7 @@ 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 = '6c5d5686515f4ea2e505b9f7dd85cbde78f51b4f'
+xla_hash = '336c9a99eeebb04a4ad9a8e6ce3b8d96e5ecd5e7'
 
 http_archive(
     name = "xla",
diff --git a/setup.py b/setup.py
index 66d6056386d..a6c86d808e3 100644
--- a/setup.py
+++ b/setup.py
@@ -64,10 +64,10 @@
 
 base_dir = os.path.dirname(os.path.abspath(__file__))
 
-_date = '20240726'
+_date = '20240801'
 _libtpu_version = f'0.1.dev{_date}'
-_libtpu_storage_path = f'https://storage.googleapis.com/libtpu-nightly-releases/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}+default-py3-none-any.whl'
-_jax_version = f'0.4.31.dev{_date}'
+_libtpu_storage_path = f'https://storage.googleapis.com/libtpu-nightly-releases/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}+nightly-py3-none-any.whl'
+_jax_version = f'0.4.32.dev{_date}'
 
 
 def _get_build_mode():