Use Mooncake in BNN doc (#521)

TuringLang · Nov 25, 2024 · 28fcc61 · 28fcc61
1 parent aca574f
commit 28fcc61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/03-bayesian-neural-network/index.qmd b/tutorials/03-bayesian-neural-network/index.qmd
@@ -19,7 +19,7 @@ using Turing
 using FillArrays
 using Lux
 using Plots
-using Tracker
+import Mooncake
 using Functors
 
 using LinearAlgebra
@@ -208,7 +208,7 @@ setprogress!(false)
 ```{julia}
 # Perform inference.
 N = 2_000
-ch = sample(bayes_nn(reduce(hcat, xs), ts), NUTS(; adtype=AutoTracker()), N);
+ch = sample(bayes_nn(reduce(hcat, xs), ts), NUTS(; adtype=AutoMooncake(; config=nothing)), N);
 ```
 
 Now we extract the parameter samples from the sampled chain as `θ` (this is of size `5000 x 20` where `5000` is the number of iterations and `20` is the number of parameters).