Skip to content

Commit

Permalink
Change deprecated jax.tree_map to avoid warnings:
Browse files Browse the repository at this point in the history 
```
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
  • Loading branch information
@copybara-github
CLU Authors authored and copybara-github committed Apr 5, 2024
1 parent 873ad23 commit 5e7a1e9
Showing 1 changed file with 18 additions and 19 deletions.
37 changes: 18 additions & 19 deletions clu_synopsis.ipynb
View file

Large diffs are not rendered by default.

0 comments on commit 5e7a1e9

Please sign in to comment.