code: def _lw_jacobian_term(z, delta):
      """Computes the jacobian term for the Lambert W transformation."""
      # See Eq (31) of https://www.hindawi.com/journals/tswj/2015/909231/.
      if delta == 0:
          return np.ones_like(z)
      return np.where(
          z == 0.0,
          np.ones_like(z),
          _w_delta_squared_numpy(z, delta)
          / (z * (1.0 + _w0(delta * z**2))),
      )
  
imports: from absl.testing import parameterized
  import numpy
  from scipy import special
  from scipy import stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import lambertw_transform
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 4 for calling a global function with example usage: _lw_jacobian_term(z, delta) and returns: np, np