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.



Data and notebooks can be found at my github repository.

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

pip install numpyro arviz causalgraphicalmodels daft