code: def eval_zero(fn: Callable, *inputs: PyTree[Any]) -> PyTree[Array]:
      out = filter_eval_shape(fn, *inputs)
      return jtu.tree_map(lambda x: jnp.zeros(x.shape, x.dtype), out)
  
file_path: equinox/equinox/internal/_misc.py
imports: from collections.abc import Callable, Sequence
  from typing import Any, TypeVar
  import jax.lax
  import jax.numpy
  import jax.tree_util
  from jaxtyping import Array, PyTree, Scalar
  from .._eval_shape import filter_eval_shape
  from .._filters import is_array
  from ._nontraceable import nonbatchable
label_desc: Use image node_id 3 for calling a global function with example usage: eval_zero(fn) and returns: jtu