Open
Description
The jax.tree_map function is deprecated from jax and jaxlib 0.4.26. This should be updated to jax.tree.map (jax v0.4.25 or newer) or jax.tree_util.tree_map (any JAX version).
Metadata
Metadata
Assignees
Labels
No labels
{{ message }}
No one assigned
No labels
No type
No projects
No milestone
None yet
No branches or pull requests