Respect gradient_backend in fit_mvn_to_MAP

jessegrabowski · jessegrabowski · commit 9d47188e6b94 · 2025-02-03T21:44:26.000+08:00
diff --git a/pymc_extras/inference/laplace.py b/pymc_extras/inference/laplace.py
@@ -552,6 +552,7 @@ def fit_laplace(
         model=model,
         on_bad_cov=on_bad_cov,
         transform_samples=fit_in_unconstrained_space,
+        gradient_backend=gradient_backend,
         zero_tol=zero_tol,
         diag_jitter=diag_jitter,
         compile_kwargs=compile_kwargs,
