From ffb07cdadb5dc3bc43485cf041dbc2b43136109e Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Mon, 9 Dec 2024 15:38:37 -0500 Subject: [PATCH] Update versions for v0.4.37 release. --- CHANGELOG.md | 4 ++++ jax/version.py | 2 +- setup.py | 2 +- 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 92dcfe6cc3f6..c2e237dd6f4f 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -12,10 +12,14 @@ When releasing, please add the new-release-boilerplate to docs/pallas/CHANGELOG. ## jax 0.4.37 +This is a patch release of jax 0.4.36. Only "jax" was released at this version. + * Bug fixes + * Fixed a bug where `jit` would error if an argument was named `f` (#25329). * Fix a bug that will throw `index out of range` error in {func}`jax.lax.while_loop` if the user register pytree node class with different aux data for the flatten and flatten_with_path. + * Pinned a new libtpu release (0.0.6) that fixes a compiler bug on TPU v6e. ## jax 0.4.36 (Dec 5, 2024) diff --git a/jax/version.py b/jax/version.py index 941b34f1226f..9da3d63f8708 100644 --- a/jax/version.py +++ b/jax/version.py @@ -21,7 +21,7 @@ import pathlib import subprocess -_version = "0.4.36" +_version = "0.4.37" # The following line is overwritten by build scripts in distributions & # releases. Do not modify this manually, or jax/jaxlib build will fail. _release_version: str | None = None diff --git a/setup.py b/setup.py index ea42d625eadc..65b07f931a71 100644 --- a/setup.py +++ b/setup.py @@ -23,7 +23,7 @@ # The following should be updated after each new jaxlib release. _latest_jaxlib_version_on_pypi = '0.4.35' -_libtpu_version = '0.0.5' +_libtpu_version = '0.0.6' _libtpu_nightly_terminal_version = '0.1.dev20241010+nightly.cleanup' def load_version_module(pkg_path):