code: def _softplus(x, name=None):  # pylint: disable=unused-argument
      if not JAX_MODE:
          # This is effectively inlining jax.nn.softplus, which is (currently)
          # defined as np.logaddexp(x, 0.).
          # Both are numerically fine (see discussion in b/146563881).
          return np.log1p(np.exp(-np.abs(x))) + np.maximum(x, 0.0)
      return jax.nn.softplus(x)
  
imports: import collections
  import functools
  import numpy
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy.numpy_array import _reverse
  from tensorflow_probability.python.internal.backend.numpy.numpy_array import one_hot
  from tensorflow_probability.python.internal.backend.numpy.ops import _convert_to_tensor
label_desc: Use image node_id 28 for calling a global function with example usage: _softplus(x, name) and returns: jax, unknown