Port hierarchical multivariate forecasting model from Pyro to NumPyro (#2006)

* nb init

* improve nb

* final run

* smaller images

* smaller image

* clear cell outputs

* smaller images

* use jax.lax.scan

* add loader function

* improve deps

* load from parts

* update nb

* remove unecessary requirements

* update paths

Author: Juan Orduz

docs/source/_static/img/tutorials/hierarchical_forecasting.png

@@ -86,6 +86,7 @@ NumPyro documentation
    examples/cvae
    tutorials/tbip
    examples/var2
+   tutorials/hierarchical_forecasting
 
 .. nbgallery::
    :maxdepth: 1