jax.nn.log1mexp

jax.nn.log1mexp = <jax._src.custom_derivatives.custom_jvp object>

Numerically stable calculation of \(\log(1 - \exp(-x))\).

This function is undefined for \(x < 0\).

Based on TensorFlow’s implementation.

References

[1]

Martin Mächler. Accurately Computing log(1 − exp(−|a|)) Assessed by the Rmpfr package..

Parameters:

x (ArrayLike)

Return type:

Array