Behrens' Bayesian Learner Model : How to share parameters between steps?

Questions
23

Can you edit the previous post to include the model code (and perhaps the data if that’s feasible)?

Edit: the plots for I and k (and the pairwise) might also be useful. Do you get any warnings / divergences during sampling?