update pin

pytorch · Dec 11, 2024 · ae2f85e · ae2f85e
1 parent 7dd2697
commit ae2f85e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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 = '0020d6a56274c8893517adb22b46a3fcb5fa340a'
+xla_hash = '30f22f4d6cb523e035c237f30aeac4e00ae34821'
 
 http_archive(
     name = "xla",

diff --git a/setup.py b/setup.py
@@ -65,9 +65,9 @@
 base_dir = os.path.dirname(os.path.abspath(__file__))
 
 USE_NIGHTLY = False  # whether to use nightly or stable libtpu and jax
-_date = '20241205'
-_libtpu_version = f'0.0.5'
-_jax_version = f'0.4.36'
+_date = '20241209'
+_libtpu_version = f'0.0.6'
+_jax_version = f'0.4.37'
 _libtpu_wheel_name = f'libtpu-{_libtpu_version}'
 
 if USE_NIGHTLY: