code: def vector_norm(x, /, *, axis=None, keepdims=False, ord=2):
      """Computes the vector norm of a vector (or batch of vectors) x."""
      return jax.numpy.linalg.vector_norm(
          x, axis=axis, keepdims=keepdims, ord=ord
      )
  
imports: from functools import partial
  from typing import NamedTuple
  import jax
  from jax.experimental.array_api._data_type_functions import _promote_to_default_dtype
label_desc: Use image node_id 23 for calling a global function with example usage: vector_norm() and returns: jax