code: def test_zero_buffer():
      def cond_fun(carry):
          step, val = carry
          return step < 5
  
      def body_fun(carry):
          step, val = carry
          return step + 1, val
  
      init_step = 0
      init_val = jnp.zeros((5, 0))
      init_carry = (init_step, init_val)
  
      def run(init_carry):
          return eqxi.while_loop(
              cond_fun,
              body_fun,
              init_carry,
              kind="checkpointed",
              max_steps=5,
          )
  
      jax.linearize(run, init_carry)
  
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 15 for calling a global function with example usage: test_zero_buffer() and returns: unknown, unknown,  val, eqxi