From Pyro to NumPyro: Forecasting Hierarchical Models - Part II

In this second notebook, we continue working on the NumPyro implementation of the hierarchical forecasting models presented in Pyro’s forecasting documentation: Forecasting III: hierarchical models . In this second part, we extend the model described in the first part From Pyro to NumPyro: Forecasting Hierarchical Models - Part I by adding all stations to the model. &#xA; &#xA; Prepare Notebook…