code: def test_symbolic_zero(capfd):
      def cond_fun(carry):
          return True
  
      def body_fun(carry):
          carry0, carry1, carry2, carry3, carry4, carry5 = carry
          return (
              carry2,
              carry3,
              lax.stop_gradient(carry1),
              carry3,
              carry5,
              carry4,
          )
  
      @jax.grad
      def run(init_carry):
          init_carry = init_carry + (5, jnp.array(6))
          outs = eqxi.while_loop(
              cond_fun,
              body_fun,
              init_carry,
              kind="checkpointed",
              max_steps=5,
          )
          return outs[0]
  
      capfd.readouterr()
      run((1.0, 2.0, jnp.array(3.0), jnp.array(4.0)))
      text, _ = capfd.readouterr()
      assert (
          "symbolic_zero_gradient "
          "(True, True, True, (False, True, False, True, True, True))"
      ) in text
  
imports: import functools
  import timeit
  from typing import Optional
  import equinox
  import equinox.internal
  import jax
  import jax.lax
  import jax.lib
  import jax.numpy
  import jax.random
  import jax.tree_util
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 16 for calling a global function with example usage: test_symbolic_zero(capfd) and returns: True, carry2,  carry3,  lax,  carry3,  carry5,  carry4, outs