Dynamic number of Dirichlet distributions?

I could have sworn I tried that… works fine once the shape is specified!

Once the shape pull request is settled, I might try to make sure that is detected automatically…