Statistical Rethinking (2nd ed.) with NumPyro

I am a fan of the book Statistical Rethinking, so I port the codes of its second edition to NumPyro. I hope that the book and this translation will be helpful not only for NumPyro/Pyro users but also for ones who are willing to do Bayesian statistics in Python.

How to read the notebooks

• Read on the site: https://fehiepsi.github.io/rethinking-numpyro/

• Use GitHub's renderer: https://github.com/fehiepsi/rethinking-numpyro/tree/master/notebooks/

• Use Jupyter's nbviewer: https://nbviewer.jupyter.org/github/fehiepsi/rethinking-numpyro/tree/master/notebooks/

Installation

The following tools are used for some analysis and visualizations: arviz for posteriors, networkx and daft for causal graphs, and (optional) ete3 for phylogenetic trees.

pip install numpyro arviz daft networkx

Excercises

If you need help solving the exercises with numpyro, there are resolutions from chapter 2 and 3 here.

Acknowledgements

Kapil Sachdeva and Manuel A. Vázquez has helped fix many code and plot mismatches since the release. Many thanks!