Prepare for jax and jaxlib 0.4.7 release

PiperOrigin-RevId: 519785176
jax-ml · Mar 27, 2023 · e9cac5e · e9cac5e
1 parent e21aee1
commit e9cac5e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -7,7 +7,7 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 #    and update the sha256 with the result.
 http_archive(
     name = "xla",
-    sha256 = "9f39af4d81d2c8bd52b47f4ef37dfd6642c6950112e4d9d3d95ae19982c46eba",
+    sha256 = "e1f99fbc3d149a8a2f83986f6ed4c0895d2901c0faf41fa6b26efc58948ac82e",
     strip_prefix = "xla-0f31407ee498e6dba242d03f8d382ebcfcc61790",
     urls = [
         "https://github.com/openxla/xla/archive/0f31407ee498e6dba242d03f8d382ebcfcc61790.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.6'
+_current_jaxlib_version = '0.4.7'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.6'
 _available_cuda11_cudnn_versions = ['82', '86']
 _default_cuda11_cudnn_version = '86'
 _default_cuda12_cudnn_version = '88'
-_libtpu_version = '0.1.dev20230309'
+_libtpu_version = '0.1.dev20230327'
 
 _dct = {}
 with open('jax/version.py', encoding='utf-8') as f: