mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Prepare for JAX release 0.5.2 (patch release over 0.5.1)

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2025-03-04 12:50:43 -08:00
parent bb80a56898
commit ce224293b1
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
CHANGELOG.md
View File

@ -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. 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) ## jax 0.5.1 (Feb 24, 2025)
* New Features * New Features

2
jax/version.py
View File

@ -21,7 +21,7 @@ import os
import pathlib import pathlib
import subprocess import subprocess
_version = "0.5.1" _version = "0.5.2"
# The following line is overwritten by build scripts in distributions & # The following line is overwritten by build scripts in distributions &
# releases. Do not modify this manually, or jax/jaxlib build will fail. # releases. Do not modify this manually, or jax/jaxlib build will fail.
_release_version: str | None = None _release_version: str | None = None