We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5e7a1e9 commit d381738Copy full SHA for d381738
CHANGELOG.md
@@ -114,3 +114,9 @@ Last release before dropping support for Python 3.8 and 3.9
114
- Removes numpy version pin
115
- Adds sharding annotations, dtype, total bytes to `parameter_overview`
116
- 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
@@ -37,7 +37,7 @@
37
38
setup(
39
name="clu",
40
- version="0.0.11",
+ version="0.0.12",
41
description="Set of libraries for ML training loops in JAX.",
42
author="Common Loop Utils Authors",
43
author_email="no-reply@google.com",
0 commit comments