code: def eval_shape(fn, *args):
      """Evaluates the shape/dtypes of fn statically."""
      return jax.eval_shape(fn, *args)
  
imports: import functools
  import jax
  from jax import lax
  from jax import random
  from jax import tree_util
  import jax.numpy
label_desc: Use image node_id 26 for calling a global function with example usage: eval_shape(fn) and returns: jax