From 7e6620a57775084dfa8d438ae4fd27f3ef365018 Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Wed, 4 Dec 2024 13:20:44 -0500 Subject: [PATCH] JAX release 0.4.36. --- jax/version.py | 2 +- setup.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/jax/version.py b/jax/version.py index 3e8a8291ec8d..941b34f1226f 100644 --- a/jax/version.py +++ b/jax/version.py @@ -137,7 +137,7 @@ def make_release_tree(self, base_dir, files): __version__ = _get_version_string() -_minimum_jaxlib_version = "0.4.35" +_minimum_jaxlib_version = "0.4.36" def _version_as_tuple(version_str): return tuple(int(i) for i in version_str.split(".") if i.isdigit()) diff --git a/setup.py b/setup.py index a3b54f7aa94f..ea42d625eadc 100644 --- a/setup.py +++ b/setup.py @@ -19,11 +19,11 @@ project_name = 'jax' -_current_jaxlib_version = '0.4.35' +_current_jaxlib_version = '0.4.36' # The following should be updated after each new jaxlib release. _latest_jaxlib_version_on_pypi = '0.4.35' -_libtpu_version = '0.0.2' +_libtpu_version = '0.0.5' _libtpu_nightly_terminal_version = '0.1.dev20241010+nightly.cleanup' def load_version_module(pkg_path):