I tried it in two new Conda environments, using pip to install the main branch of the pymc GitHub and pymc==4.0.0b4 just like in the Colab notebook. I also used Python 3.10.2 now. However, the same error still comes up. I will try it on a different OS now.