mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36: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.
-->
## 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

2
jax/version.py
View File

@ -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