[ar1_turningpts] Removed PyMC and changed the style (#584)
* Update ar1_turningpts.md
* Update ar1_turningpts.md
* Update ar1_turningpts.md
* Update ar1_turningpts.md
* fix typos and add explanations
* fix typo
* check grammar and style
* use Greek letters and arviz plot
* set numpyro.set_host_device_count
* modify var_nam in arviz
* Reorder numpyro imports and device count setup
* test local_device_count
* update the vectorization strategy
* fix typo and add explanation
* address review comments
* fix document refrence
* Fix ar1_turningpts NumPyro/JAX rewrite: run on latest libs
Bring PR #584 to a mergeable, style-compliant state. Verified end-to-end
on a GPU box (numpyro NUTS + JAX), ~110s, all 5 figures render.
Blocking fixes:
- Correct `NamedTuplep` import typo that broke the whole build
- Vectorize the NumPyro likelihood (was a per-observation Python loop
creating one sample site per datum) to match ar1_bayes.md
- Fix plot_extended_Wecker plotting an undefined `future_path` (it fell
back to the true-parameter global); now plots the posterior-draw paths
- Adapt to arviz 1.x: az.plot_trace -> az.plot_trace_dist (new API drops
data=/compact=/lines=/backend_kwargs=); update prose for the new
left-column densities / right-column traces layout
Style-guide compliance (QuantEcon.manual):
- Add create_ar1() factory with parameter validation
- @jax.jit the hot compute_path_statistics
- lw=2 on line charts; tidy figure captions; revert header to .myst
- Restore the true-parameters vs posterior legend on the comparison plot
- Remove orphaned `colors` variable
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
---------
Co-authored-by: mmcky <mamckay@gmail.com>
Co-authored-by: Humphrey Yang <u6474961@anu.edu.au>
Co-authored-by: Matt McKay <mmcky@users.noreply.github.com>
Co-authored-by: John Stachurski <john.stachurski@gmail.com>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
1 file changed
Lines changed: 503 additions & 293 deletions