I have some speed ups for hierarchical models in the repo here. All are only a two-level hierarchy, and I suspect you’ll really only benefit from the use of columns_dot_product rather than rows_dot_product. I suspect that your likelihood loops can be vectorized, and thereby reduce_sum’d, but I’d have to think about how to go about that.
1 Like