code: def _log_diff(x, y):
      return logsumexp(
          jnp.array([x, y]),
          b=jnp.array([jnp.ones_like(x), -jnp.ones_like(y)]),
          axis=0,
      )
  
file_path: jax/jax/_src/scipy/stats/truncnorm.py
imports: import scipy.stats
  from jax import lax
  import jax.numpy
  from jax._src.numpy.util import _wraps, promote_args_inexact
  from jax._src.scipy.stats import norm
  from jax._src.scipy.special import logsumexp, log_ndtr, ndtr
label_desc: Use image node_id 1 for calling a global function with example usage: _log_diff(x, y) and returns: logsumexp