Skip to content

Commit d381738

Browse files
Marvin182copybara-github
authored andcommitted
Release v0.0.12.
PiperOrigin-RevId: 623238257
1 parent 5e7a1e9 commit d381738

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

CHANGELOG.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,3 +114,9 @@ Last release before dropping support for Python 3.8 and 3.9
114114
- Removes numpy version pin
115115
- Adds sharding annotations, dtype, total bytes to `parameter_overview`
116116
- Makes `clu.metrics.Std` support same shapes as `clu.metrics.Average`
117+
118+
## v0.0.12
119+
120+
- Switch from `jax.tree_map` (deprecated since JAX 0.4.26) to
121+
`jax.tree_util.tree_map`.
122+
- Improvements to parameter overview.

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737

3838
setup(
3939
name="clu",
40-
version="0.0.11",
40+
version="0.0.12",
4141
description="Set of libraries for ML training loops in JAX.",
4242
author="Common Loop Utils Authors",
4343
author_email="no-reply@google.com",

0 commit comments

Comments
 (0)