This really helpful thanks - I didn’t know pip provides a slower sampling speed compared to conda. I’ll give mamba a shot and provide an update here.
Also, with your suggestion on updating to the latest pytensor, unfortunately, because of software restrictions, I have to stick with the version pytensor-rel-2.17.2 since installing a higher version requires microsoft visual build tools c++ 14.0 or greater which I’m not able to install.