Change initialization defaults to include randomness

Author: jessegrabowski

python/nutpie/compile_pymc.py

@@ -145,16 +145,16 @@ def with_data(self, **updates):
             user_data=user_data,
         )
 
-    def _make_sampler(self, settings, cores, progress_type):
-        model = self._make_model()
+    def _make_sampler(self, settings, init_mean, cores, progress_type):
+        model = self._make_model(init_mean)
         return _lib.PySampler.from_pymc(
             settings,
             cores,
             model,
             progress_type,
         )
 
-    def _make_model(self):
+    def _make_model(self, init_mean):
         expand_fn = _lib.ExpandFunc(
             self.n_dim,
             self.n_expanded,
@@ -434,7 +434,7 @@ def compile_pymc_model(
     gradient_backend: Literal["pytensor", "jax"] = "pytensor",
     overrides: dict[Union["Variable", str], np.ndarray | float | int] | None = None,
     jitter_rvs: set["TensorVariable"] | None = None,
-    default_strategy: Literal["support_point", "prior"] = "support_point",
+    default_strategy: Literal["support_point", "prior"] = "prior",
     **kwargs,
 ) -> CompiledModel:
     """Compile necessary functions for sampling a pymc model.
@@ -469,6 +469,9 @@ def compile_pymc_model(
             "and restart your kernel in case you are in an interactive session."
         )
 
+    if default_strategy == "support_point" and jitter_rvs is None:
+        jitter_rvs = set(model.free_RVs)
+
     initial_point_fn = make_initial_point_fn(
         model=model,
         overrides=overrides,