Code conversion Bayesian Modeling and Computation in Python ch6. Help on AR(1) model implementation

Thank you @junpenglao. Any idea how if not when Pymc4 will start to work in google colab?