Plotting NUTS trace

[Storopoli]

I think there are some mistakes in your hierarchical model.
Where is idx?

Try to see some examples here: https://storopoli.io/Bayesian-Julia/pages/10_multilevel_models/