From 70b4477296dcfb82b5ef1b25606a4e76d5013859 Mon Sep 17 00:00:00 2001 From: Yash Katariya Date: Tue, 7 May 2024 07:00:34 -0700 Subject: [PATCH] Start jax and jaxlib 0.4.27 release PiperOrigin-RevId: 631409685 --- CHANGELOG.md | 4 ---- setup.py | 4 ++-- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index d5d931bb71e4..49f76b163b0f 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -25,10 +25,6 @@ Remember to align the itemized text with the first line of an item within a list to the callback. * `complex_arr.astype(bool)` now follows the same semantics as NumPy, returning False where `complex_arr` is equal to `0 + 0j`, and True otherwise. - * Async dispatch expensive computations on the CPU backend. This only applies - to non-parallel computations, as we already do async dispatch for parallel - computations. You can recover the old behavior by setting - `jax.config.update('jax_cpu_enable_async_dispatch', False)`. * `core.Token` now is a non-trivial class which wraps a `jax.Array`. It could be created and threaded in and out of computations to build up dependency. The singleton object `core.token` has been removed, users now should create diff --git a/setup.py b/setup.py index 0d4f555c36c2..fe2ed831f9e9 100644 --- a/setup.py +++ b/setup.py @@ -22,12 +22,12 @@ project_name = 'jax' -_current_jaxlib_version = '0.4.26' +_current_jaxlib_version = '0.4.27' # The following should be updated with each new jaxlib release. _latest_jaxlib_version_on_pypi = '0.4.26' _default_cuda12_cudnn_version = '89' _available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version] -_libtpu_version = '0.1.dev20240403' +_libtpu_version = '0.1.dev20240507' def load_version_module(pkg_path): spec = importlib.util.spec_from_file_location(