diff --git a/CHANGELOG.md b/CHANGELOG.md index e5c1903d1..07910c638 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -14,6 +14,13 @@ Remember to align the itemized text with the first line of an item within a list When releasing, please add the new-release-boilerplate to docs/pallas/CHANGELOG.md. --> +## jax 0.5.2 (Mar 4, 2025) + +Patch release of 0.5.1 + +* Bug fixes + * Fixes TPU metric logging and `tpu-info`, which was broken in 0.5.1 + ## jax 0.5.1 (Feb 24, 2025) * New Features diff --git a/jax/version.py b/jax/version.py index 1aa049434..616950577 100644 --- a/jax/version.py +++ b/jax/version.py @@ -21,7 +21,7 @@ import os import pathlib import subprocess -_version = "0.5.1" +_version = "0.5.2" # 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