FYI: I submitted a PR so this would be easier to implement: https://github.com/pymc-devs/pymc3/pull/3070