1
0
mirror of https://github.com/ROCm/jax.git synced 2025-04-20 05:46:06 +00:00

Merge branch 'release/0.5.2' into main

This commit is contained in:
Skye Wanderman-Milne 2025-03-04 18:47:20 -08:00
commit a6c858f04b

@ -23,6 +23,13 @@ When releasing, please add the new-release-boilerplate to docs/pallas/CHANGELOG.
true, matching the current behavior. If set to false, JAX does not need to
emit code clamping negative indices, which improves code size.
## 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