From 038dfeec150af0046afe6ced9ea52392f8a1aecf Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Wed, 8 May 2024 18:55:53 +0000 Subject: [PATCH] Prepare 0.4.28 release. --- CHANGELOG.md | 15 +++++++++++++++ setup.py | 4 ++-- 2 files changed, 17 insertions(+), 2 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 1224335a2308..26035e478559 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -8,6 +8,9 @@ Remember to align the itemized text with the first line of an item within a list ## jax 0.4.28 +* Bug fixes + * Reverted a change to `make_jaxpr` that was breaking Equinox (#21116). + * Deprecations & removals * The ``kind`` argument to {func}`jax.numpy.sort` and {func}`jax.numpy.argsort` is now removed. Use `stable=True` or `stable=False` instead. @@ -15,9 +18,21 @@ Remember to align the itemized text with the first line of an item within a list module. Use the ``compute_capability`` attribute of a GPU device, returned by {func}`jax.devices` or {func}`jax.local_devices`, instead. +* Changes + * The minimum jaxlib version of this release is 0.4.27. + ## jaxlib 0.4.28 + +* Bug fixes * Fixes a memory corruption bug in the type name of Array and JIT Python objects in Python 3.10 or earlier. + * Fixed a warning `'+ptx84' is not a recognized feature for this target` + under CUDA 12.4. + * Fixed a slow compilation problem on CPU. + +* Changes + * The Windows build is now built with Clang instead of MSVC. + ## jax 0.4.27 (May 7, 2024) diff --git a/setup.py b/setup.py index 9b01e3b289b6..56061b821c2a 100644 --- a/setup.py +++ b/setup.py @@ -22,12 +22,12 @@ project_name = 'jax' -_current_jaxlib_version = '0.4.27' +_current_jaxlib_version = '0.4.28' # The following should be updated with each new jaxlib release. _latest_jaxlib_version_on_pypi = '0.4.27' _default_cuda12_cudnn_version = '89' _available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version] -_libtpu_version = '0.1.dev20240507' +_libtpu_version = '0.1.dev20240508' def load_version_module(pkg_path): spec = importlib.util.spec_from_file_location(