code: def logpdf(
      x: ArrayLike, loc: ArrayLike = 0, scale: ArrayLike = 1
  ) -> Array:
      x, loc, scale = promote_args_inexact(
          "uniform.logpdf", x, loc, scale
      )
      log_probs = lax.neg(lax.log(scale))
      return where(
          logical_or(lax.gt(x, lax.add(loc, scale)), lax.lt(x, loc)),
          -inf,
          log_probs,
      )
  
file_path: jax/jax/_src/scipy/stats/uniform.py
imports: import scipy.stats
  from jax import lax
  from jax import numpy
  from jax.numpy import where, inf, logical_or
  from jax._src.typing import Array, ArrayLike
  from jax._src.numpy.util import _wraps, promote_args_inexact
label_desc: Use image node_id 1 for calling a global function with example usage: logpdf(x, loc, scale) and returns: where