Vesselin Grantcharov

Update levenberg_marquardt.py

1

Retracing can be avoided by simply adding the tf.function decorator sample showing the effect of the change on runtime and warning that is mentioned without decorator: https://gist.github.com/Vesalon/a643c355d3caa2bf30f6ecf485505276

use `jax.scipy.special` for printing in jax

4

Sympy expressions with, for example, the `erf` function which get printed to jax or numpy will have the `erf` call translated to `math.erf`, but scipy will use the scipy version...