Numerical Issues with StickBreaking in ADVI

I replaced the StickBreaking with something that is less convoluted and should be numerically fine: https://github.com/pymc-devs/pymc3/pull/3620

However, the problem stays the same.