Jax and jaxlib 0.3.18 release candidate.

abaelhe · Sep 26, 2022 · bcd36d8 · bcd36d8
1 parent 53de057
commit bcd36d8
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -8,8 +8,8 @@ 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.18 (Unreleased)
-* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.17...main).
+## jax 0.3.18 (Sep 26, 2022)
+* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.17...jax-v0.3.18).
 * Changes
   * Ahead-of-time lowering and compilation functionality (tracked in
     {jax-issue}`#7733`) is stable and public. See [the
@@ -22,6 +22,9 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
     `jax.soft_pmap` is undocumented. If it were documented, a deprecation period
     would have been provided.
 
+## jaxlib 0.3.18 (Sep 26, 2022)
+* [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.15...jaxlib-v0.3.18).
+
 ## jax 0.3.17 (Aug 31, 2022)
 * [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.16...jax-v0.3.17).
 * Bugs
@@ -62,9 +65,6 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
     implementation switched on by default, meaning the old implementation is
     deprecated; see [JEP 11830](https://jax.readthedocs.io/en/latest/jep/11830-new-remat-checkpoint.html).
 
-## jaxlib 0.3.16 (Unreleased)
-* [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.15...main).
-
 ## jax 0.3.15 (July 22, 2022)
 * [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.14...jax-v0.3.15).
 * Changes

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 = "org_tensorflow",
-    sha256 = "9073ab3cbf3a89baee459f6e953cee240864393774f568fdba200a6ff5512c9f",
-    strip_prefix = "tensorflow-a4905aa04186bcaf89b06032baa450cc1ce103ad",
+    sha256 = "23e1cf573ee3857f4e598dd14ae2c1554d72fbf6bcec500c6315aa930865eddf",
+    strip_prefix = "tensorflow-8dcaf6b98a6a49c85eb470140ba8506e71a3b5af",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/a4905aa04186bcaf89b06032baa450cc1ce103ad.tar.gz",
+        "https://github.com/tensorflow/tensorflow/archive/8dcaf6b98a6a49c85eb470140ba8506e71a3b5af.tar.gz",
     ],
 )
 

diff --git a/setup.py b/setup.py
@@ -19,14 +19,14 @@
 
 from setuptools import setup, find_packages
 
-_current_jaxlib_version = '0.3.15'
+_current_jaxlib_version = '0.3.18'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.3.15'
 _available_cuda_versions = ['11']
 _default_cuda_version = '11'
 _available_cudnn_versions = ['82', '805']
 _default_cudnn_version = '82'
-_libtpu_version = '0.1.dev20220723'
+_libtpu_version = '0.1.dev20220926'
 
 _dct = {}
 with open('jax/version.py') as f: