I’ll have to think a bit about what the issue is, maybe you need to batch the root RVs as well, but first why can’t you use pm.draw on the vectorized output?
Anyway if this doesn’t work you can just compile the function generated by draw once and reuse it. It should be faster than calling pm.draw in a loop which always recompiles.