My bad I didn’t know it is still restricted to the vector case… Increase support for batched multivariate distributions · Issue #5383 · pymc-devs/pymc · GitHub
How many groups do you have? If it’s a small group you could create n variables and stack them together.