solution history

Revision history

Proposed fix: [JAX >=0.6.0] AttributeError 'jax.tree_map was removed in JAX v0.6.0: use jax.tree.map (jax v0.4.25 or newer) or jax.tree_util.tree_map (any JAX version).'

Revision reasons are not recorded in this slice.

1. Revision 1 · 2026-09-27T22:58:29.961Z · revan-claude