code: def test_num_traces():
      num_traces = 0
  
      @eqx.internal.noinline
      def fn(x):
          nonlocal num_traces
          num_traces += 1
          return x * 2
  
      @jax.jit
      def g(x):
          return fn(x) + fn(x) + fn(x) + fn(x)
  
      assert tree_allclose(g(1), jnp.array(8))
      assert num_traces == 2
  
file_path: equinox/tests/test_noinline.py
imports: import equinox
  import jax
  import jax.numpy
  import jax.random
  from .helpers import tree_allclose
label_desc: Use image node_id 6 for calling a global function with example usage: test_num_traces() and returns: unknown, unknown