use jax stable version

pytorch · Sep 27, 2024 · 756be91 · 756be91
1 parent 324cf2c
commit 756be91
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -50,8 +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 = '32ebd694c4d0442e241d76324ff1a721831366b4'
-xla_hash = '06bbcd1a798cd49bb811674fbed8823dfef51cc4'
+xla_hash = '590cd6fcd1ed24ab9cf494789a0fc524b94a4a6a'
 
 http_archive(
     name = "xla",

diff --git a/setup.py b/setup.py
@@ -64,7 +64,7 @@
 
 base_dir = os.path.dirname(os.path.abspath(__file__))
 
-_date = '20240926'
+_date = '20240916'
 _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}+nightly-py3-none-any.whl'
 _jax_version = f'0.4.33.dev{_date}'