Merge pull request #11187 from sharadmv:release

PiperOrigin-RevId: 456338899

CHANGELOG.md

@@ -8,7 +8,11 @@ Remember to align the itemized text with the first line of an item within a list
 PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
 -->
 
-## jax 0.3.14 (Unreleased)
+## jax 0.3.15 (Unreleased)
+
+## jaxlib 0.3.15 (Unreleased)
+
+## jax 0.3.14 (June 21, 2022)
 * [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.13...main).
 * Breaking changes
   * {func}`jax.experimental.compilation_cache.initialize_cache` does not support
@@ -57,7 +61,7 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
     bucket path as input.
   * Added {func}`jax.scipy.stats.gennorm`.
 
-## jaxlib 0.3.11 (Unreleased)
+## jaxlib 0.3.14 (June 21, 2022)
 * [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.10...main).
   * x86-64 Mac wheels now require Mac OS 10.14 (Mojave) or newer. Mac OS 10.14
     was released in 2018, so this should not be a very onerous requirement.

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 = "org_tensorflow",
-    sha256 = "db36e99e992dc44c59f6630070c50d1cb85ea229bcb3bc30b45d23c15a9cd0ea",
-    strip_prefix = "tensorflow-ea9e3b4fb5640294479b517a7efc493e8f9d02f2",
+    sha256 = "bb0770fab82bdb243e0118671cb056253c1c74fc46fb64cc9c4c7d8ce2773c29",
+    strip_prefix = "tensorflow-0550a551f158167fa88acb17261d96e959529f70",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/ea9e3b4fb5640294479b517a7efc493e8f9d02f2.tar.gz",
+        "https://github.com/tensorflow/tensorflow/archive/0550a551f158167fa88acb17261d96e959529f70.tar.gz",
     ],
 )
 

jaxlib/version.py

@@ -17,4 +17,4 @@
 # reflect the most recent available binaries.
 # __version__ should be increased after releasing the current version
 # (i.e. on main, this is always the next version to be released).
-__version__ = "0.3.11"
+__version__ = "0.3.14"

setup.py

@@ -14,14 +14,14 @@
 
 from setuptools import setup, find_packages
 
-_current_jaxlib_version = '0.3.10'
+_current_jaxlib_version = '0.3.14'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.3.10'
 _available_cuda_versions = ['11']
 _default_cuda_version = '11'
 _available_cudnn_versions = ['82', '805']
 _default_cudnn_version = '82'
-_libtpu_version = '0.1.dev20220504'
+_libtpu_version = '0.1.dev20220621'
 
 _dct = {}
 with open('jax/version.py') as f: