code: def l2_norm(x):
      """Compute the l2 norm of a vector."""
      return jnp.sqrt((x * x).sum(axis=-1))
  
imports: import functools
  from typing import Optional, Sequence
  from flax import linen
  import jax
  import jax.numpy
  from scenic.projects.knowledge_visual_language.models import constants
  from t5x.examples.t5 import layers
  from t5x.examples.t5 import network
label_desc: Use image node_id 3 for calling a global function with example usage: l2_norm(x) and returns: jnp