af2bind icon indicating copy to clipboard operation
af2bind copied to clipboard
verified publisher icon sokrypton

Fixed JAX Tree error

Open AtharvaTilewale opened this issue 2 months ago • 0 comments

Fixed the issue "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)."

AtharvaTilewale avatar Nov 11 '25 15:11 AtharvaTilewale