Thank you!
I think it would be good software engineering practice to have an API that is not dependent on the specific model, but I am not sure what is the preferred way to do this as I am not very deep into pymc3. I will come up with something and we continue the discussion in the PR 