I’m experiencing the same issue and I need my model to run quickly the first time I run it. Is there anything I could manually cache before running pm.sample() to make it faster? The issue is not the sampling itself, but rather what pymc3 is doing behind the hood before it even assigns the sampler.
I’ve tried running the model as a Python script right after installing all the packages in a fresh environment, but it did not help.