code: def pmean(xs, axis_name):
      warnings.warn("use jax.lax.pmean instead", DeprecationWarning)
      return lax.pmean(xs, axis_name)
  
imports: import collections
  import itertools
  import warnings
  from collections.abc import Iterable
  import jax
  import jax.numpy
  import numpy
  from jax import core, lax
  from jax.extend import linear_util
  from jax.interpreters import partial_eval
label_desc: Use image node_id 4 for calling a global function with example usage: pmean(xs, axis_name) and returns: lax