See this thread. I’m in the process of adding some more code to the linked Github repository and I somewhat concur with @aleshing’s claim that the model, as implemented, is wrong. It’s just a special case where c_i = 1 (see paper linked in @aleshing’s post in the other thread), which admittedly is not the most interesting case.
If you wanna help out, please give me a shout.