code: def duplicate_params(
      params: jnp.ndarray, repeats: int, ma_training: bool
  ) -> jnp.ndarray:
      if ma_training:
          return jnp.tile(params, (repeats,) + (1,) * (params.ndim - 1))
      else:
          return jnp.repeat(params, repeats=repeats, axis=0)
  
imports: import logging
  from functools import partial
  from typing import Tuple
  from typing import Union
  import time
  import jax
  import jax.numpy
  from jax import random
  from jax.tree_util import tree_map
  from evojax.obs_norm import ObsNormalizer
  from evojax.task.base import TaskState
  from evojax.task.base import VectorizedTask
  from evojax.policy.base import PolicyState
  from evojax.policy.base import PolicyNetwork
  from evojax.util import create_logger
label_desc: Use image node_id 6 for calling a global function with example usage: duplicate_params(params, repeats, ma_training) and returns: jnp, jnp