ott.math.utils.lambertw

ott.math.utils.lambertw(z, tol=1e-08, max_iter=100) = <jax._src.custom_derivatives.custom_jvp object>

Principal branch of the Lambert W function.

This implementation uses Halley’s iteration and the global initialization proposed in [Iacono and Boyd, 2017], Eq. 20 .

Parameters:

• z (Array) – Array.

• tol (float) – Tolerance threshold.

• max_iter (int) – Maximum number of iterations.

Return type:

Array

Returns:

The Lambert W evaluated at z.