Merge pull request jax-ml#16869 from skye:version

PiperOrigin-RevId: 551628701
harborn · Jul 27, 2023 · c75e85d · c75e85d
2 parents a480aa8 + 0b24b2b
commit c75e85d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -7,10 +7,10 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 #    and update the sha256 with the result.
 http_archive(
     name = "xla",
-    sha256 = "082fd460b11e9a672d4b1cab99807128958094dc56c8fff9ea40ad19fe1a313f",
-    strip_prefix = "xla-ed7886d05343d2a787829a380b723a675499067c",
+    sha256 = "60558f5c2a2ff46f46194febdc808180b6536a9d42b2835cd738afff462a496b",
+    strip_prefix = "xla-81b357b48e9b1c9dc3381a6ab4a2a64b06c34fc7",
     urls = [
-        "https://github.com/openxla/xla/archive/ed7886d05343d2a787829a380b723a675499067c.tar.gz",
+        "https://github.com/openxla/xla/archive/81b357b48e9b1c9dc3381a6ab4a2a64b06c34fc7.tar.gz",
     ],
 )
 

diff --git a/setup.py b/setup.py
@@ -19,13 +19,13 @@
 
 from setuptools import setup, find_packages
 
-_current_jaxlib_version = '0.4.13'
+_current_jaxlib_version = '0.4.14'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.13'
 _available_cuda11_cudnn_versions = ['86']
 _default_cuda11_cudnn_version = '86'
 _default_cuda12_cudnn_version = '89'
-_libtpu_version = '0.1.dev20230622'
+_libtpu_version = '0.1.dev20230727'
 
 _dct = {}
 with open('jax/version.py', encoding='utf-8') as f: