code: def zeros_creator(next_creator, shape, dtype, init, context):
      del context
      init = jnp.zeros
      return next_creator(shape, dtype, init)
  
file_path: dm-haiku/haiku/_src/eval_shape.py
imports: from unittest import mock
  from haiku._src import base
  from haiku._src import basic
  from haiku._src import stateful
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling a global function with example usage: zeros_creator(next_creator, shape, dtype, init, context) and returns: next_creator