Btw, just ran this code on Google Collab, it took less than 2 minutes. Not great, but a big improvement that is making me think that I may have installed PyMC wrong (which is weird, since I followed the recommended conda way of installing it). I will start with a fresh ubuntu, conda and pymc installations and see if I get better speed.