search

pyro-ppl / numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
https://num.pyro.ai
Apache License 2.0
2.15k stars 234 forks source link

Make figure in AR2 example reproducible #1816

Closed damonbayer closed 3 months ago

damonbayer commented 3 months ago

This PR adds back in code to produce the ar2 plot in the ar2 example. This is based on the code that was removed in https://github.com/pyro-ppl/numpyro/pull/1350. It was last available in https://github.com/pyro-ppl/numpyro/blob/c5b1a9229daf964a98eb12a8ad905e0d0a6c9437/examples/ar2.py.

The figure was generated with:

python ar2.py --num-samples 1000 --num-warmup 1000 --num-chains 4

Closes https://github.com/pyro-ppl/numpyro/issues/1815.

damonbayer commented 3 months ago

CI checks should now pass. Sorry about that.

damonbayer commented 3 months ago

@fehiepsi I don't think any of the test failures are related to my contribution. Please let me know if you disagree.

fehiepsi commented 3 months ago

Yes they are unrelated - @juanitorduz is working on them in #1817

fehiepsi commented 3 months ago

@damonbayer Could you sync with master? They are fixed upstream.

damonbayer commented 3 months ago

@fehiepsi Fork is synced!