Puma's collapsed Dirichlet categorical distribution is asymptotically slower to simulate than Lite's

[axch]

because it does a linear scan of the aux+weight vector instead of Lite's tree-based sampling algorithm. Implementing The Plan (#453) and porting it to Puma should fix this.

[axch]

Marking blocked on #453.