code: def top1_accuracy_tiebreak(
      logits: chex.Array,
      targets: chex.Array,
      *,
      rng: chex.PRNGKey,
      eps: float = 1e-6,
  ) -> chex.Array:
      """Compute the top-1 accuracy with an argmax of targets (random tie-break)."""
      noise = jax.random.uniform(
          rng, shape=targets.shape, minval=-eps, maxval=eps
      )
      acc = jnp.argmax(logits, axis=-1) == jnp.argmax(
          targets + noise, axis=-1
      )
      return jnp.mean(acc)
  
file_path: acme/acme/agents/jax/mpo/rollout_loss.py
imports: from typing import Tuple
  from acme import types
  from acme.adders import reverb
  from acme.agents.jax.mpo import categorical_mpo
  from acme.agents.jax.mpo import networks
  from acme.agents.jax.mpo import types
  from acme.agents.jax.mpo import utils
  from acme.jax import networks
  import chex
  import jax
  import jax.numpy
  import rlax
label_desc: Use image node_id 2 for calling a global function with example usage: top1_accuracy_tiebreak(logits, targets) and returns: jnp