Issue opened: Type instability with Dirichlet distribution · Issue #1276 · TuringLang/Turing.jl · GitHub
And yeah, I wouldn’t be surprised to see it run a bit slower when it needs to sample the weights as well, but the slowdown with the type instability is ~25x, which did get my attention.