Jmat.Real.lambertw, newton loop step

Percentage Accurate: 77.9% → 98.0%
Time: 7.7s
Alternatives: 7
Speedup: 313.0×

Specification

?
\[\begin{array}{l} \\ \begin{array}{l} t_0 := wj \cdot e^{wj}\\ wj - \frac{t_0 - x}{e^{wj} + t_0} \end{array} \end{array} \]
(FPCore (wj x)
 :precision binary64
 (let* ((t_0 (* wj (exp wj)))) (- wj (/ (- t_0 x) (+ (exp wj) t_0)))))
double code(double wj, double x) {
	double t_0 = wj * exp(wj);
	return wj - ((t_0 - x) / (exp(wj) + t_0));
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    real(8) :: t_0
    t_0 = wj * exp(wj)
    code = wj - ((t_0 - x) / (exp(wj) + t_0))
end function
public static double code(double wj, double x) {
	double t_0 = wj * Math.exp(wj);
	return wj - ((t_0 - x) / (Math.exp(wj) + t_0));
}
def code(wj, x):
	t_0 = wj * math.exp(wj)
	return wj - ((t_0 - x) / (math.exp(wj) + t_0))
function code(wj, x)
	t_0 = Float64(wj * exp(wj))
	return Float64(wj - Float64(Float64(t_0 - x) / Float64(exp(wj) + t_0)))
end
function tmp = code(wj, x)
	t_0 = wj * exp(wj);
	tmp = wj - ((t_0 - x) / (exp(wj) + t_0));
end
code[wj_, x_] := Block[{t$95$0 = N[(wj * N[Exp[wj], $MachinePrecision]), $MachinePrecision]}, N[(wj - N[(N[(t$95$0 - x), $MachinePrecision] / N[(N[Exp[wj], $MachinePrecision] + t$95$0), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]]
\begin{array}{l}

\\
\begin{array}{l}
t_0 := wj \cdot e^{wj}\\
wj - \frac{t_0 - x}{e^{wj} + t_0}
\end{array}
\end{array}

Sampling outcomes in binary64 precision:

Local Percentage Accuracy vs ?

The average percentage accuracy by input value. Horizontal axis shows value of an input variable; the variable is choosen in the title. Vertical axis is accuracy; higher is better. Red represent the original program, while blue represents Herbie's suggestion. These can be toggled with buttons below the plot. The line is an average while dots represent individual samples.

Accuracy vs Speed?

Herbie found 7 alternatives:

AlternativeAccuracySpeedup
The accuracy (vertical axis) and speed (horizontal axis) of each alternatives. Up and to the right is better. The red square shows the initial program, and each blue circle shows an alternative.The line shows the best available speed-accuracy tradeoffs.

Initial Program: 77.9% accurate, 1.0× speedup?

\[\begin{array}{l} \\ \begin{array}{l} t_0 := wj \cdot e^{wj}\\ wj - \frac{t_0 - x}{e^{wj} + t_0} \end{array} \end{array} \]
(FPCore (wj x)
 :precision binary64
 (let* ((t_0 (* wj (exp wj)))) (- wj (/ (- t_0 x) (+ (exp wj) t_0)))))
double code(double wj, double x) {
	double t_0 = wj * exp(wj);
	return wj - ((t_0 - x) / (exp(wj) + t_0));
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    real(8) :: t_0
    t_0 = wj * exp(wj)
    code = wj - ((t_0 - x) / (exp(wj) + t_0))
end function
public static double code(double wj, double x) {
	double t_0 = wj * Math.exp(wj);
	return wj - ((t_0 - x) / (Math.exp(wj) + t_0));
}
def code(wj, x):
	t_0 = wj * math.exp(wj)
	return wj - ((t_0 - x) / (math.exp(wj) + t_0))
function code(wj, x)
	t_0 = Float64(wj * exp(wj))
	return Float64(wj - Float64(Float64(t_0 - x) / Float64(exp(wj) + t_0)))
end
function tmp = code(wj, x)
	t_0 = wj * exp(wj);
	tmp = wj - ((t_0 - x) / (exp(wj) + t_0));
end
code[wj_, x_] := Block[{t$95$0 = N[(wj * N[Exp[wj], $MachinePrecision]), $MachinePrecision]}, N[(wj - N[(N[(t$95$0 - x), $MachinePrecision] / N[(N[Exp[wj], $MachinePrecision] + t$95$0), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]]
\begin{array}{l}

\\
\begin{array}{l}
t_0 := wj \cdot e^{wj}\\
wj - \frac{t_0 - x}{e^{wj} + t_0}
\end{array}
\end{array}

Alternative 1: 98.0% accurate, 0.6× speedup?

\[\begin{array}{l} \\ \begin{array}{l} t_0 := x \cdot -4 + x \cdot 1.5\\ t_1 := wj \cdot e^{wj}\\ \mathbf{if}\;wj + \frac{x - t_1}{e^{wj} + t_1} \leq 2 \cdot 10^{-14}:\\ \;\;\;\;{wj}^{3} \cdot \left(\left(\left(-1 - -2 \cdot t_0\right) - x \cdot -3\right) - x \cdot 0.6666666666666666\right) + \left(\left(1 - t_0\right) \cdot {wj}^{2} + \left(x + -2 \cdot \left(wj \cdot x\right)\right)\right)\\ \mathbf{else}:\\ \;\;\;\;wj + \frac{t_1 - x}{e^{wj}} \cdot \frac{-1}{wj + 1}\\ \end{array} \end{array} \]
(FPCore (wj x)
 :precision binary64
 (let* ((t_0 (+ (* x -4.0) (* x 1.5))) (t_1 (* wj (exp wj))))
   (if (<= (+ wj (/ (- x t_1) (+ (exp wj) t_1))) 2e-14)
     (+
      (*
       (pow wj 3.0)
       (- (- (- -1.0 (* -2.0 t_0)) (* x -3.0)) (* x 0.6666666666666666)))
      (+ (* (- 1.0 t_0) (pow wj 2.0)) (+ x (* -2.0 (* wj x)))))
     (+ wj (* (/ (- t_1 x) (exp wj)) (/ -1.0 (+ wj 1.0)))))))
double code(double wj, double x) {
	double t_0 = (x * -4.0) + (x * 1.5);
	double t_1 = wj * exp(wj);
	double tmp;
	if ((wj + ((x - t_1) / (exp(wj) + t_1))) <= 2e-14) {
		tmp = (pow(wj, 3.0) * (((-1.0 - (-2.0 * t_0)) - (x * -3.0)) - (x * 0.6666666666666666))) + (((1.0 - t_0) * pow(wj, 2.0)) + (x + (-2.0 * (wj * x))));
	} else {
		tmp = wj + (((t_1 - x) / exp(wj)) * (-1.0 / (wj + 1.0)));
	}
	return tmp;
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    real(8) :: t_0
    real(8) :: t_1
    real(8) :: tmp
    t_0 = (x * (-4.0d0)) + (x * 1.5d0)
    t_1 = wj * exp(wj)
    if ((wj + ((x - t_1) / (exp(wj) + t_1))) <= 2d-14) then
        tmp = ((wj ** 3.0d0) * ((((-1.0d0) - ((-2.0d0) * t_0)) - (x * (-3.0d0))) - (x * 0.6666666666666666d0))) + (((1.0d0 - t_0) * (wj ** 2.0d0)) + (x + ((-2.0d0) * (wj * x))))
    else
        tmp = wj + (((t_1 - x) / exp(wj)) * ((-1.0d0) / (wj + 1.0d0)))
    end if
    code = tmp
end function
public static double code(double wj, double x) {
	double t_0 = (x * -4.0) + (x * 1.5);
	double t_1 = wj * Math.exp(wj);
	double tmp;
	if ((wj + ((x - t_1) / (Math.exp(wj) + t_1))) <= 2e-14) {
		tmp = (Math.pow(wj, 3.0) * (((-1.0 - (-2.0 * t_0)) - (x * -3.0)) - (x * 0.6666666666666666))) + (((1.0 - t_0) * Math.pow(wj, 2.0)) + (x + (-2.0 * (wj * x))));
	} else {
		tmp = wj + (((t_1 - x) / Math.exp(wj)) * (-1.0 / (wj + 1.0)));
	}
	return tmp;
}
def code(wj, x):
	t_0 = (x * -4.0) + (x * 1.5)
	t_1 = wj * math.exp(wj)
	tmp = 0
	if (wj + ((x - t_1) / (math.exp(wj) + t_1))) <= 2e-14:
		tmp = (math.pow(wj, 3.0) * (((-1.0 - (-2.0 * t_0)) - (x * -3.0)) - (x * 0.6666666666666666))) + (((1.0 - t_0) * math.pow(wj, 2.0)) + (x + (-2.0 * (wj * x))))
	else:
		tmp = wj + (((t_1 - x) / math.exp(wj)) * (-1.0 / (wj + 1.0)))
	return tmp
function code(wj, x)
	t_0 = Float64(Float64(x * -4.0) + Float64(x * 1.5))
	t_1 = Float64(wj * exp(wj))
	tmp = 0.0
	if (Float64(wj + Float64(Float64(x - t_1) / Float64(exp(wj) + t_1))) <= 2e-14)
		tmp = Float64(Float64((wj ^ 3.0) * Float64(Float64(Float64(-1.0 - Float64(-2.0 * t_0)) - Float64(x * -3.0)) - Float64(x * 0.6666666666666666))) + Float64(Float64(Float64(1.0 - t_0) * (wj ^ 2.0)) + Float64(x + Float64(-2.0 * Float64(wj * x)))));
	else
		tmp = Float64(wj + Float64(Float64(Float64(t_1 - x) / exp(wj)) * Float64(-1.0 / Float64(wj + 1.0))));
	end
	return tmp
end
function tmp_2 = code(wj, x)
	t_0 = (x * -4.0) + (x * 1.5);
	t_1 = wj * exp(wj);
	tmp = 0.0;
	if ((wj + ((x - t_1) / (exp(wj) + t_1))) <= 2e-14)
		tmp = ((wj ^ 3.0) * (((-1.0 - (-2.0 * t_0)) - (x * -3.0)) - (x * 0.6666666666666666))) + (((1.0 - t_0) * (wj ^ 2.0)) + (x + (-2.0 * (wj * x))));
	else
		tmp = wj + (((t_1 - x) / exp(wj)) * (-1.0 / (wj + 1.0)));
	end
	tmp_2 = tmp;
end
code[wj_, x_] := Block[{t$95$0 = N[(N[(x * -4.0), $MachinePrecision] + N[(x * 1.5), $MachinePrecision]), $MachinePrecision]}, Block[{t$95$1 = N[(wj * N[Exp[wj], $MachinePrecision]), $MachinePrecision]}, If[LessEqual[N[(wj + N[(N[(x - t$95$1), $MachinePrecision] / N[(N[Exp[wj], $MachinePrecision] + t$95$1), $MachinePrecision]), $MachinePrecision]), $MachinePrecision], 2e-14], N[(N[(N[Power[wj, 3.0], $MachinePrecision] * N[(N[(N[(-1.0 - N[(-2.0 * t$95$0), $MachinePrecision]), $MachinePrecision] - N[(x * -3.0), $MachinePrecision]), $MachinePrecision] - N[(x * 0.6666666666666666), $MachinePrecision]), $MachinePrecision]), $MachinePrecision] + N[(N[(N[(1.0 - t$95$0), $MachinePrecision] * N[Power[wj, 2.0], $MachinePrecision]), $MachinePrecision] + N[(x + N[(-2.0 * N[(wj * x), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision], N[(wj + N[(N[(N[(t$95$1 - x), $MachinePrecision] / N[Exp[wj], $MachinePrecision]), $MachinePrecision] * N[(-1.0 / N[(wj + 1.0), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]]]]
\begin{array}{l}

\\
\begin{array}{l}
t_0 := x \cdot -4 + x \cdot 1.5\\
t_1 := wj \cdot e^{wj}\\
\mathbf{if}\;wj + \frac{x - t_1}{e^{wj} + t_1} \leq 2 \cdot 10^{-14}:\\
\;\;\;\;{wj}^{3} \cdot \left(\left(\left(-1 - -2 \cdot t_0\right) - x \cdot -3\right) - x \cdot 0.6666666666666666\right) + \left(\left(1 - t_0\right) \cdot {wj}^{2} + \left(x + -2 \cdot \left(wj \cdot x\right)\right)\right)\\

\mathbf{else}:\\
\;\;\;\;wj + \frac{t_1 - x}{e^{wj}} \cdot \frac{-1}{wj + 1}\\


\end{array}
\end{array}
Derivation
  1. Split input into 2 regimes
  2. if (-.f64 wj (/.f64 (-.f64 (*.f64 wj (exp.f64 wj)) x) (+.f64 (exp.f64 wj) (*.f64 wj (exp.f64 wj))))) < 2e-14

    1. Initial program 75.3%

      \[wj - \frac{wj \cdot e^{wj} - x}{e^{wj} + wj \cdot e^{wj}} \]
    2. Taylor expanded in wj around 0 99.3%

      \[\leadsto \color{blue}{-1 \cdot \left(\left(0.6666666666666666 \cdot x + \left(-3 \cdot x + \left(1 + -2 \cdot \left(-4 \cdot x + 1.5 \cdot x\right)\right)\right)\right) \cdot {wj}^{3}\right) + \left(\left(1 - \left(-4 \cdot x + 1.5 \cdot x\right)\right) \cdot {wj}^{2} + \left(-2 \cdot \left(wj \cdot x\right) + x\right)\right)} \]

    if 2e-14 < (-.f64 wj (/.f64 (-.f64 (*.f64 wj (exp.f64 wj)) x) (+.f64 (exp.f64 wj) (*.f64 wj (exp.f64 wj)))))

    1. Initial program 98.1%

      \[wj - \frac{wj \cdot e^{wj} - x}{e^{wj} + wj \cdot e^{wj}} \]
    2. Step-by-step derivation
      1. *-un-lft-identity98.1%

        \[\leadsto wj - \frac{\color{blue}{1 \cdot \left(wj \cdot e^{wj} - x\right)}}{e^{wj} + wj \cdot e^{wj}} \]
      2. distribute-rgt1-in99.4%

        \[\leadsto wj - \frac{1 \cdot \left(wj \cdot e^{wj} - x\right)}{\color{blue}{\left(wj + 1\right) \cdot e^{wj}}} \]
      3. times-frac99.4%

        \[\leadsto wj - \color{blue}{\frac{1}{wj + 1} \cdot \frac{wj \cdot e^{wj} - x}{e^{wj}}} \]
    3. Applied egg-rr99.4%

      \[\leadsto wj - \color{blue}{\frac{1}{wj + 1} \cdot \frac{wj \cdot e^{wj} - x}{e^{wj}}} \]
  3. Recombined 2 regimes into one program.
  4. Final simplification99.3%

    \[\leadsto \begin{array}{l} \mathbf{if}\;wj + \frac{x - wj \cdot e^{wj}}{e^{wj} + wj \cdot e^{wj}} \leq 2 \cdot 10^{-14}:\\ \;\;\;\;{wj}^{3} \cdot \left(\left(\left(-1 - -2 \cdot \left(x \cdot -4 + x \cdot 1.5\right)\right) - x \cdot -3\right) - x \cdot 0.6666666666666666\right) + \left(\left(1 - \left(x \cdot -4 + x \cdot 1.5\right)\right) \cdot {wj}^{2} + \left(x + -2 \cdot \left(wj \cdot x\right)\right)\right)\\ \mathbf{else}:\\ \;\;\;\;wj + \frac{wj \cdot e^{wj} - x}{e^{wj}} \cdot \frac{-1}{wj + 1}\\ \end{array} \]

Alternative 2: 97.7% accurate, 1.4× speedup?

\[\begin{array}{l} \\ \begin{array}{l} \mathbf{if}\;wj \leq -1.02 \cdot 10^{-8}:\\ \;\;\;\;wj + \frac{wj \cdot e^{wj} - x}{e^{wj}} \cdot \frac{-1}{wj + 1}\\ \mathbf{else}:\\ \;\;\;\;\left(1 - \left(x \cdot -4 + x \cdot 1.5\right)\right) \cdot {wj}^{2} + \left(x + -2 \cdot \left(wj \cdot x\right)\right)\\ \end{array} \end{array} \]
(FPCore (wj x)
 :precision binary64
 (if (<= wj -1.02e-8)
   (+ wj (* (/ (- (* wj (exp wj)) x) (exp wj)) (/ -1.0 (+ wj 1.0))))
   (+
    (* (- 1.0 (+ (* x -4.0) (* x 1.5))) (pow wj 2.0))
    (+ x (* -2.0 (* wj x))))))
double code(double wj, double x) {
	double tmp;
	if (wj <= -1.02e-8) {
		tmp = wj + ((((wj * exp(wj)) - x) / exp(wj)) * (-1.0 / (wj + 1.0)));
	} else {
		tmp = ((1.0 - ((x * -4.0) + (x * 1.5))) * pow(wj, 2.0)) + (x + (-2.0 * (wj * x)));
	}
	return tmp;
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    real(8) :: tmp
    if (wj <= (-1.02d-8)) then
        tmp = wj + ((((wj * exp(wj)) - x) / exp(wj)) * ((-1.0d0) / (wj + 1.0d0)))
    else
        tmp = ((1.0d0 - ((x * (-4.0d0)) + (x * 1.5d0))) * (wj ** 2.0d0)) + (x + ((-2.0d0) * (wj * x)))
    end if
    code = tmp
end function
public static double code(double wj, double x) {
	double tmp;
	if (wj <= -1.02e-8) {
		tmp = wj + ((((wj * Math.exp(wj)) - x) / Math.exp(wj)) * (-1.0 / (wj + 1.0)));
	} else {
		tmp = ((1.0 - ((x * -4.0) + (x * 1.5))) * Math.pow(wj, 2.0)) + (x + (-2.0 * (wj * x)));
	}
	return tmp;
}
def code(wj, x):
	tmp = 0
	if wj <= -1.02e-8:
		tmp = wj + ((((wj * math.exp(wj)) - x) / math.exp(wj)) * (-1.0 / (wj + 1.0)))
	else:
		tmp = ((1.0 - ((x * -4.0) + (x * 1.5))) * math.pow(wj, 2.0)) + (x + (-2.0 * (wj * x)))
	return tmp
function code(wj, x)
	tmp = 0.0
	if (wj <= -1.02e-8)
		tmp = Float64(wj + Float64(Float64(Float64(Float64(wj * exp(wj)) - x) / exp(wj)) * Float64(-1.0 / Float64(wj + 1.0))));
	else
		tmp = Float64(Float64(Float64(1.0 - Float64(Float64(x * -4.0) + Float64(x * 1.5))) * (wj ^ 2.0)) + Float64(x + Float64(-2.0 * Float64(wj * x))));
	end
	return tmp
end
function tmp_2 = code(wj, x)
	tmp = 0.0;
	if (wj <= -1.02e-8)
		tmp = wj + ((((wj * exp(wj)) - x) / exp(wj)) * (-1.0 / (wj + 1.0)));
	else
		tmp = ((1.0 - ((x * -4.0) + (x * 1.5))) * (wj ^ 2.0)) + (x + (-2.0 * (wj * x)));
	end
	tmp_2 = tmp;
end
code[wj_, x_] := If[LessEqual[wj, -1.02e-8], N[(wj + N[(N[(N[(N[(wj * N[Exp[wj], $MachinePrecision]), $MachinePrecision] - x), $MachinePrecision] / N[Exp[wj], $MachinePrecision]), $MachinePrecision] * N[(-1.0 / N[(wj + 1.0), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision], N[(N[(N[(1.0 - N[(N[(x * -4.0), $MachinePrecision] + N[(x * 1.5), $MachinePrecision]), $MachinePrecision]), $MachinePrecision] * N[Power[wj, 2.0], $MachinePrecision]), $MachinePrecision] + N[(x + N[(-2.0 * N[(wj * x), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]]
\begin{array}{l}

\\
\begin{array}{l}
\mathbf{if}\;wj \leq -1.02 \cdot 10^{-8}:\\
\;\;\;\;wj + \frac{wj \cdot e^{wj} - x}{e^{wj}} \cdot \frac{-1}{wj + 1}\\

\mathbf{else}:\\
\;\;\;\;\left(1 - \left(x \cdot -4 + x \cdot 1.5\right)\right) \cdot {wj}^{2} + \left(x + -2 \cdot \left(wj \cdot x\right)\right)\\


\end{array}
\end{array}
Derivation
  1. Split input into 2 regimes
  2. if wj < -1.02000000000000003e-8

    1. Initial program 60.7%

      \[wj - \frac{wj \cdot e^{wj} - x}{e^{wj} + wj \cdot e^{wj}} \]
    2. Step-by-step derivation
      1. *-un-lft-identity60.7%

        \[\leadsto wj - \frac{\color{blue}{1 \cdot \left(wj \cdot e^{wj} - x\right)}}{e^{wj} + wj \cdot e^{wj}} \]
      2. distribute-rgt1-in89.2%

        \[\leadsto wj - \frac{1 \cdot \left(wj \cdot e^{wj} - x\right)}{\color{blue}{\left(wj + 1\right) \cdot e^{wj}}} \]
      3. times-frac89.1%

        \[\leadsto wj - \color{blue}{\frac{1}{wj + 1} \cdot \frac{wj \cdot e^{wj} - x}{e^{wj}}} \]
    3. Applied egg-rr89.1%

      \[\leadsto wj - \color{blue}{\frac{1}{wj + 1} \cdot \frac{wj \cdot e^{wj} - x}{e^{wj}}} \]

    if -1.02000000000000003e-8 < wj

    1. Initial program 82.6%

      \[wj - \frac{wj \cdot e^{wj} - x}{e^{wj} + wj \cdot e^{wj}} \]
    2. Taylor expanded in wj around 0 98.5%

      \[\leadsto \color{blue}{\left(1 - \left(-4 \cdot x + 1.5 \cdot x\right)\right) \cdot {wj}^{2} + \left(-2 \cdot \left(wj \cdot x\right) + x\right)} \]
  3. Recombined 2 regimes into one program.
  4. Final simplification98.2%

    \[\leadsto \begin{array}{l} \mathbf{if}\;wj \leq -1.02 \cdot 10^{-8}:\\ \;\;\;\;wj + \frac{wj \cdot e^{wj} - x}{e^{wj}} \cdot \frac{-1}{wj + 1}\\ \mathbf{else}:\\ \;\;\;\;\left(1 - \left(x \cdot -4 + x \cdot 1.5\right)\right) \cdot {wj}^{2} + \left(x + -2 \cdot \left(wj \cdot x\right)\right)\\ \end{array} \]

Alternative 3: 96.0% accurate, 2.6× speedup?

\[\begin{array}{l} \\ \left(1 - \left(x \cdot -4 + x \cdot 1.5\right)\right) \cdot {wj}^{2} + \left(x + -2 \cdot \left(wj \cdot x\right)\right) \end{array} \]
(FPCore (wj x)
 :precision binary64
 (+ (* (- 1.0 (+ (* x -4.0) (* x 1.5))) (pow wj 2.0)) (+ x (* -2.0 (* wj x)))))
double code(double wj, double x) {
	return ((1.0 - ((x * -4.0) + (x * 1.5))) * pow(wj, 2.0)) + (x + (-2.0 * (wj * x)));
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    code = ((1.0d0 - ((x * (-4.0d0)) + (x * 1.5d0))) * (wj ** 2.0d0)) + (x + ((-2.0d0) * (wj * x)))
end function
public static double code(double wj, double x) {
	return ((1.0 - ((x * -4.0) + (x * 1.5))) * Math.pow(wj, 2.0)) + (x + (-2.0 * (wj * x)));
}
def code(wj, x):
	return ((1.0 - ((x * -4.0) + (x * 1.5))) * math.pow(wj, 2.0)) + (x + (-2.0 * (wj * x)))
function code(wj, x)
	return Float64(Float64(Float64(1.0 - Float64(Float64(x * -4.0) + Float64(x * 1.5))) * (wj ^ 2.0)) + Float64(x + Float64(-2.0 * Float64(wj * x))))
end
function tmp = code(wj, x)
	tmp = ((1.0 - ((x * -4.0) + (x * 1.5))) * (wj ^ 2.0)) + (x + (-2.0 * (wj * x)));
end
code[wj_, x_] := N[(N[(N[(1.0 - N[(N[(x * -4.0), $MachinePrecision] + N[(x * 1.5), $MachinePrecision]), $MachinePrecision]), $MachinePrecision] * N[Power[wj, 2.0], $MachinePrecision]), $MachinePrecision] + N[(x + N[(-2.0 * N[(wj * x), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]
\begin{array}{l}

\\
\left(1 - \left(x \cdot -4 + x \cdot 1.5\right)\right) \cdot {wj}^{2} + \left(x + -2 \cdot \left(wj \cdot x\right)\right)
\end{array}
Derivation
  1. Initial program 82.0%

    \[wj - \frac{wj \cdot e^{wj} - x}{e^{wj} + wj \cdot e^{wj}} \]
  2. Taylor expanded in wj around 0 96.5%

    \[\leadsto \color{blue}{\left(1 - \left(-4 \cdot x + 1.5 \cdot x\right)\right) \cdot {wj}^{2} + \left(-2 \cdot \left(wj \cdot x\right) + x\right)} \]
  3. Final simplification96.5%

    \[\leadsto \left(1 - \left(x \cdot -4 + x \cdot 1.5\right)\right) \cdot {wj}^{2} + \left(x + -2 \cdot \left(wj \cdot x\right)\right) \]

Alternative 4: 95.9% accurate, 28.5× speedup?

\[\begin{array}{l} \\ \left(x + -2 \cdot \left(wj \cdot x\right)\right) + wj \cdot wj \end{array} \]
(FPCore (wj x) :precision binary64 (+ (+ x (* -2.0 (* wj x))) (* wj wj)))
double code(double wj, double x) {
	return (x + (-2.0 * (wj * x))) + (wj * wj);
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    code = (x + ((-2.0d0) * (wj * x))) + (wj * wj)
end function
public static double code(double wj, double x) {
	return (x + (-2.0 * (wj * x))) + (wj * wj);
}
def code(wj, x):
	return (x + (-2.0 * (wj * x))) + (wj * wj)
function code(wj, x)
	return Float64(Float64(x + Float64(-2.0 * Float64(wj * x))) + Float64(wj * wj))
end
function tmp = code(wj, x)
	tmp = (x + (-2.0 * (wj * x))) + (wj * wj);
end
code[wj_, x_] := N[(N[(x + N[(-2.0 * N[(wj * x), $MachinePrecision]), $MachinePrecision]), $MachinePrecision] + N[(wj * wj), $MachinePrecision]), $MachinePrecision]
\begin{array}{l}

\\
\left(x + -2 \cdot \left(wj \cdot x\right)\right) + wj \cdot wj
\end{array}
Derivation
  1. Initial program 82.0%

    \[wj - \frac{wj \cdot e^{wj} - x}{e^{wj} + wj \cdot e^{wj}} \]
  2. Taylor expanded in wj around 0 96.5%

    \[\leadsto \color{blue}{\left(1 - \left(-4 \cdot x + 1.5 \cdot x\right)\right) \cdot {wj}^{2} + \left(-2 \cdot \left(wj \cdot x\right) + x\right)} \]
  3. Taylor expanded in x around 0 96.2%

    \[\leadsto \color{blue}{{wj}^{2}} + \left(-2 \cdot \left(wj \cdot x\right) + x\right) \]
  4. Step-by-step derivation
    1. unpow296.2%

      \[\leadsto \color{blue}{wj \cdot wj} + \left(-2 \cdot \left(wj \cdot x\right) + x\right) \]
  5. Simplified96.2%

    \[\leadsto \color{blue}{wj \cdot wj} + \left(-2 \cdot \left(wj \cdot x\right) + x\right) \]
  6. Final simplification96.2%

    \[\leadsto \left(x + -2 \cdot \left(wj \cdot x\right)\right) + wj \cdot wj \]

Alternative 5: 95.3% accurate, 62.6× speedup?

\[\begin{array}{l} \\ x + wj \cdot wj \end{array} \]
(FPCore (wj x) :precision binary64 (+ x (* wj wj)))
double code(double wj, double x) {
	return x + (wj * wj);
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    code = x + (wj * wj)
end function
public static double code(double wj, double x) {
	return x + (wj * wj);
}
def code(wj, x):
	return x + (wj * wj)
function code(wj, x)
	return Float64(x + Float64(wj * wj))
end
function tmp = code(wj, x)
	tmp = x + (wj * wj);
end
code[wj_, x_] := N[(x + N[(wj * wj), $MachinePrecision]), $MachinePrecision]
\begin{array}{l}

\\
x + wj \cdot wj
\end{array}
Derivation
  1. Initial program 82.0%

    \[wj - \frac{wj \cdot e^{wj} - x}{e^{wj} + wj \cdot e^{wj}} \]
  2. Taylor expanded in wj around 0 96.5%

    \[\leadsto \color{blue}{\left(1 - \left(-4 \cdot x + 1.5 \cdot x\right)\right) \cdot {wj}^{2} + \left(-2 \cdot \left(wj \cdot x\right) + x\right)} \]
  3. Taylor expanded in x around 0 96.2%

    \[\leadsto \color{blue}{{wj}^{2}} + \left(-2 \cdot \left(wj \cdot x\right) + x\right) \]
  4. Step-by-step derivation
    1. unpow296.2%

      \[\leadsto \color{blue}{wj \cdot wj} + \left(-2 \cdot \left(wj \cdot x\right) + x\right) \]
  5. Simplified96.2%

    \[\leadsto \color{blue}{wj \cdot wj} + \left(-2 \cdot \left(wj \cdot x\right) + x\right) \]
  6. Taylor expanded in wj around 0 95.5%

    \[\leadsto wj \cdot wj + \color{blue}{x} \]
  7. Final simplification95.5%

    \[\leadsto x + wj \cdot wj \]

Alternative 6: 4.3% accurate, 313.0× speedup?

\[\begin{array}{l} \\ wj \end{array} \]
(FPCore (wj x) :precision binary64 wj)
double code(double wj, double x) {
	return wj;
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    code = wj
end function
public static double code(double wj, double x) {
	return wj;
}
def code(wj, x):
	return wj
function code(wj, x)
	return wj
end
function tmp = code(wj, x)
	tmp = wj;
end
code[wj_, x_] := wj
\begin{array}{l}

\\
wj
\end{array}
Derivation
  1. Initial program 82.0%

    \[wj - \frac{wj \cdot e^{wj} - x}{e^{wj} + wj \cdot e^{wj}} \]
  2. Taylor expanded in wj around inf 3.8%

    \[\leadsto \color{blue}{wj} \]
  3. Final simplification3.8%

    \[\leadsto wj \]

Alternative 7: 84.3% accurate, 313.0× speedup?

\[\begin{array}{l} \\ x \end{array} \]
(FPCore (wj x) :precision binary64 x)
double code(double wj, double x) {
	return x;
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    code = x
end function
public static double code(double wj, double x) {
	return x;
}
def code(wj, x):
	return x
function code(wj, x)
	return x
end
function tmp = code(wj, x)
	tmp = x;
end
code[wj_, x_] := x
\begin{array}{l}

\\
x
\end{array}
Derivation
  1. Initial program 82.0%

    \[wj - \frac{wj \cdot e^{wj} - x}{e^{wj} + wj \cdot e^{wj}} \]
  2. Taylor expanded in wj around 0 84.9%

    \[\leadsto \color{blue}{x} \]
  3. Final simplification84.9%

    \[\leadsto x \]

Developer target: 78.8% accurate, 1.5× speedup?

\[\begin{array}{l} \\ wj - \left(\frac{wj}{wj + 1} - \frac{x}{e^{wj} + wj \cdot e^{wj}}\right) \end{array} \]
(FPCore (wj x)
 :precision binary64
 (- wj (- (/ wj (+ wj 1.0)) (/ x (+ (exp wj) (* wj (exp wj)))))))
double code(double wj, double x) {
	return wj - ((wj / (wj + 1.0)) - (x / (exp(wj) + (wj * exp(wj)))));
}
real(8) function code(wj, x)
    real(8), intent (in) :: wj
    real(8), intent (in) :: x
    code = wj - ((wj / (wj + 1.0d0)) - (x / (exp(wj) + (wj * exp(wj)))))
end function
public static double code(double wj, double x) {
	return wj - ((wj / (wj + 1.0)) - (x / (Math.exp(wj) + (wj * Math.exp(wj)))));
}
def code(wj, x):
	return wj - ((wj / (wj + 1.0)) - (x / (math.exp(wj) + (wj * math.exp(wj)))))
function code(wj, x)
	return Float64(wj - Float64(Float64(wj / Float64(wj + 1.0)) - Float64(x / Float64(exp(wj) + Float64(wj * exp(wj))))))
end
function tmp = code(wj, x)
	tmp = wj - ((wj / (wj + 1.0)) - (x / (exp(wj) + (wj * exp(wj)))));
end
code[wj_, x_] := N[(wj - N[(N[(wj / N[(wj + 1.0), $MachinePrecision]), $MachinePrecision] - N[(x / N[(N[Exp[wj], $MachinePrecision] + N[(wj * N[Exp[wj], $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]
\begin{array}{l}

\\
wj - \left(\frac{wj}{wj + 1} - \frac{x}{e^{wj} + wj \cdot e^{wj}}\right)
\end{array}

Reproduce

?
herbie shell --seed 2023174 
(FPCore (wj x)
  :name "Jmat.Real.lambertw, newton loop step"
  :precision binary64

  :herbie-target
  (- wj (- (/ wj (+ wj 1.0)) (/ x (+ (exp wj) (* wj (exp wj))))))

  (- wj (/ (- (* wj (exp wj)) x) (+ (exp wj) (* wj (exp wj))))))