I’ll need to do some rewriting before sharing it here, but sure. However, can you think of any reason why it would work with PyMC 3.1 and not PyMC 3.2? I think the main point here is this difference, which is also observed with the extremely simple model above on AWS: version 3.1 works while 3.2 does not