diff --git a/WORKSPACE b/WORKSPACE index 3869f82a3d3..e0539e0e6ba 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -50,7 +50,7 @@ new_local_repository( # curl -L https://github.com/openxla/xla/archive/.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 index 0f2025ac4bd..9c9e772385f 100644 --- 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: