code: def logpdf(
      x: ArrayLike, loc: ArrayLike = 0, scale: ArrayLike = 1
  ) -> Array:
      x, loc, scale = promote_args_inexact("norm.logpdf", x, loc, scale)
      scale_sqrd = lax.square(scale)
      log_normalizer = lax.log(
          lax.mul(_lax_const(x, 2 * np.pi), scale_sqrd)
      )
      quadratic = lax.div(lax.square(lax.sub(x, loc)), scale_sqrd)
      return lax.div(
          lax.add(log_normalizer, quadratic), _lax_const(x, -2)
      )
  
file_path: jax/jax/_src/scipy/stats/norm.py
imports: from typing import cast
  import numpy
  import scipy.stats
  from jax import lax
  import jax.numpy
  from jax._src.lax.lax import _const
  from jax._src.numpy.util import _wraps, promote_args_inexact
  from jax._src.typing import Array, ArrayLike
  from jax.scipy import special
label_desc: Use image node_id 1 for calling a global function with example usage: logpdf(x, loc, scale) and returns: lax