Commit 5e7a1e9

CLU Authors

authored and

copybara-github

committed

Change deprecated jax.tree_map to avoid warnings:

``` DeprecationWarning: jax.tree_map is deprecated: use jax.tree.map (jax v0.4.25 or newer) or jax.tree_util.tree_map (any JAX version). ``` PiperOrigin-RevId: 622276898

1 parent 873ad23 commit 5e7a1e9Copy full SHA for 5e7a1e9

1 file changed

+18

-19

lines changed

clu_synopsis.ipynb

1 file changed

+18

-19

lines changed

`‎clu_synopsis.ipynb‎`

Large diffs are not rendered by default.

Comments

(0)