code: def pdf(x: ArrayLike, p: ArrayLike) -> Array:
      return lax.exp(logpdf(x, p))
  
imports: import scipy.stats
  from jax import lax
  from jax._src.numpy.util import _wraps, promote_args_inexact
  from jax._src.typing import Array, ArrayLike
label_desc: Use image node_id 3 for calling a global function with example usage: pdf(x, p) and returns: lax