code: def test_jaxpr2jaxpr_custom_idempotent():
      def fn(x):
          x = jnp.invert(x)
          x = eqxi.unvmap_any(x)
          x = jnp.invert(x)
          return x
  
      jaxpr = cast(jax.core.ClosedJaxpr, jax.make_jaxpr(fn)(True))
      jaxpr2 = eqxi.finalise_jaxpr(jaxpr)
      jaxpr3 = eqxi.finalise_jaxpr(jaxpr2)
      _assert_jaxpr_equal(jaxpr2, jaxpr3)
  
      jaxpr = jax.make_jaxpr(jax.vmap(fn))(jnp.array([True, False]))
      jaxpr = cast(jax.core.ClosedJaxpr, jaxpr)
      jaxpr2 = eqxi.finalise_jaxpr(jaxpr)
      jaxpr3 = eqxi.finalise_jaxpr(jaxpr2)
      _assert_jaxpr_equal(jaxpr2, jaxpr3)
  
file_path: equinox/tests/test_finalise_jaxpr.py
imports: from typing import cast
  import equinox.internal
  import jax
  import jax.core
  import jax.lax
  import jax.numpy
  from .helpers import tree_allclose
label_desc: Use image node_id 5 for calling a global function with example usage: test_jaxpr2jaxpr_custom_idempotent() and returns: x