Slow sampling speed

I am not sure why things are running slow for you, but I’m working on something similar and am using pm.math.dot instead of pyt.tensor.dot. Not sure if that will make an improvement for you