tfp.substrates.jax.math.sqrt1pm1

Compute sqrt(x + 1) - 1 elementwise in a numerically stable way.

x Float Tensor.

sqrt1pm1 Float Tensor of sqrt1pm1(x).