scale FD derivatives by tangent norm

PlasmaControl · Jul 26, 2024 · aa570d4 · aa570d4
1 parent 0b2207f
commit aa570d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/desc/utils.py b/desc/utils.py
@@ -766,7 +766,7 @@ def func_jvp(primals, tangents):
             def f(x):
                 return jax.flatten_util.ravel_pytree(func(*unflatx(x)))[0]
 
-            tangent_out = (f(x + fd_step * vh) - y) / fd_step
+            tangent_out = (f(x + fd_step * vh) - y) / fd_step * normv
             tangent_out = unflaty(tangent_out)
 
             return primal_out, tangent_out