refactor: jacobian should use tensorize

Michal-Novomestsky · Michal-Novomestsky · commit f20abb9ddce9 · 2025-08-12T20:00:48.000+10:00
diff --git a/pytensor/tensor/optimize.py b/pytensor/tensor/optimize.py
@@ -560,7 +560,7 @@ def L_op(self, inputs, outputs, output_grads):
         implicit_f = grad(inner_fx, inner_x)
 
         df_dx, *df_dtheta_columns = jacobian(
-            implicit_f, [inner_x, *inner_args], disconnected_inputs="ignore"
+            implicit_f, [inner_x, *inner_args], disconnected_inputs="ignore", vectorize=True
         )
         grad_wrt_args = implict_optimization_grads(
             df_dx=df_dx,

Original file line number	Diff line number	Diff line change
`@@ -560,7 +560,7 @@ def L_op(self, inputs, outputs, output_grads):`
`560`	`560`	`implicit_f = grad(inner_fx, inner_x)`
`561`	`561`
`562`	`562`	`df_dx, *df_dtheta_columns = jacobian(`
`563`		`- implicit_f, [inner_x, *inner_args], disconnected_inputs="ignore"`
	`563`	`+ implicit_f, [inner_x, *inner_args], disconnected_inputs="ignore", vectorize=True`
`564`	`564`	`)`
`565`	`565`	`grad_wrt_args = implict_optimization_grads(`
`566`	`566`	`df_dx=df_dx,`