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

After the limitation of the new user, I am back :wink:

Here is the plot.

As a comparison, ‪Michael Waskom (The author of seaborn) implement the model by using numpy. Here is the result of him. His code can find here