This looks really cool and useful.
I think that useful aspect about such a ModelBuilder class could be to enable users to integrate pymc models into scikit-learn pipeline to combine them with transformers and stuff.
To make that possible, the .fit() and .predict() would need to match the typical scikit-learn signature. Have you considered writing the function in that way?