set vectorized jac flag to true

Michal-Novomestsky · Michal-Novomestsky · commit b3a33514ef45 · 2025-09-21T17:33:36.000+10:00
diff --git a/pymc_extras/model/marginal/distributions.py b/pymc_extras/model/marginal/distributions.py
@@ -464,6 +464,7 @@ def get_laplace_approx(
     x0, _ = minimize(
         objective=-logp_objective,  # logp(x | y, params) = logp(y | x, params) + logp(x | params) + const (const omitted during minimization)
         x=x,
+        use_vectorized_jac=True,
         **minimizer_kwargs,
     )
 

Original file line number	Diff line number	Diff line change
`@@ -464,6 +464,7 @@ def get_laplace_approx(`
`464`	`464`	`x0, _ = minimize(`
`465`	`465`	`objective=-logp_objective, # logp(x \| y, params) = logp(y \| x, params) + logp(x \| params) + const (const omitted during minimization)`
`466`	`466`	`x=x,`
	`467`	`+ use_vectorized_jac=True,`
`467`	`468`	`**minimizer_kwargs,`
`468`	`469`	`)`
`469`	`470`