Numerical Issues with StickBreaking in ADVI

At this point, it looks to me more like a problem of stickbreaking.
I knew that I am missing one transformation that is also producing simplex :sweat_smile: