tfp.experimental.substrates.numpy.math.lambertw