code: def l2_normalize(
      tensor: Array, axis: int = -1, epsilon: float = 1e-6
  ):
      """L2 normalize an input tensor."""
  
      return tensor / jnp.linalg.norm(
          tensor, axis=axis, keepdims=True + epsilon
      )
  
imports: from typing import Optional, Tuple, Union
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: l2_normalize(tensor, axis, epsilon) and returns: unknown