set numpyro.set_host_device_count

xuanguang-li · xuanguang-li · commit 56d96a968f9f · 2025-09-05T16:41:37.000+08:00
diff --git a/lectures/ar1_turningpts.md b/lectures/ar1_turningpts.md
@@ -66,6 +66,7 @@ import arviz as az
 sns.set_style('white')
 colors = sns.color_palette()
 key = random.PRNGKey(0)
+numpyro.set_host_device_count(4)
 ```
 
 ## A Univariate First-Order Autoregressive Process
