Release v0.0.12.

PiperOrigin-RevId: 623238257
google · Apr 9, 2024 · d381738 · d381738
1 parent 5e7a1e9
commit d381738
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -114,3 +114,9 @@ Last release before dropping support for Python 3.8 and 3.9
 -   Removes numpy version pin
 -   Adds sharding annotations, dtype, total bytes to `parameter_overview`
 -   Makes `clu.metrics.Std` support same shapes as `clu.metrics.Average`
+
+## v0.0.12
+
+-   Switch from `jax.tree_map` (deprecated since JAX 0.4.26) to
+    `jax.tree_util.tree_map`.
+-   Improvements to parameter overview.
diff --git a/setup.py b/setup.py
@@ -37,7 +37,7 @@
 
 setup(
     name="clu",
-    version="0.0.11",
+    version="0.0.12",
     description="Set of libraries for ML training loops in JAX.",
     author="Common Loop Utils Authors",
     author_email="[email protected]",