I’ve got a bare-bones implementation of a vector-ready AR distribution on my fork. You can see it here: [WIP] VAR update by ckrapu · Pull Request #4666 · pymc-devs/pymc3 · GitHub.
All you would have to do is make sure you have a rho which is 3 dimensional, with the last two dimensions being square.