code: def tile_nested_array(
      nest: acme_types.NestedArray, num: int, axis: int
  ):
      def _tile_array(array: jnp.ndarray) -> jnp.ndarray:
          reps = [1] * array.ndim
          reps[axis] = num
          return jnp.tile(array, reps)
  
      return jax.tree_map(_tile_array, nest)
  
imports: from typing import Any, Mapping, NamedTuple, Tuple
  from acme import specs
  from acme import types
  from acme.agents.jax import actor_core
  from acme.jax import networks
  from acme.jax import types
  from acme.jax import utils
  from acme.tf import utils
  import jax
  import jax.numpy
  import reverb
  import tensorflow
  import tree
label_desc: Use image node_id 1 for calling a global function with example usage: tile_nested_array(nest, num, axis) and returns: jax, jnp