Remove jit.

JaxGaussianProcesses · Nov 28, 2023 · af7bb11 · af7bb11
1 parent cf300fd
commit af7bb11
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/examples/graph_kernels.py b/docs/examples/graph_kernels.py
@@ -172,9 +172,9 @@
 # %%
 opt_posterior, training_history = gpx.fit_scipy(
     model=posterior,
-    objective=jit(gpx.ConjugateMLL(negative=True)),
+    objective=gpx.ConjugateMLL(negative=True),
     train_data=D,
-    max_iters=1000
+    max_iters=1000,
 )
 
 # %% [markdown]