Module: tfp.experimental.substrates.jax.math.special

Implements special functions in TensorFlow.

Functions

bessel_iv_ratio(...): None

lambertw(...): None

lambertw_winitzki_approx(...): Computes Winitzki approximation to Lambert W function at z >= -1/exp(1).

lbeta(...): Returns log(Beta(x, y)).

log_gamma_correction(...): Returns the error of the Stirling approximation to lgamma(x) for x >= 8.

log_gamma_difference(...): Returns lgamma(y) - lgamma(x + y), accurately.

round_exponential_bump_function(...): Function supported on [-1, 1], smooth on the real line, with a round top.