Any suggestions on speeding up the model?

Thank you so much for the detailed explanation. I am trying to find a function which can help me to find the reason on a slow sampling. In my understanding, the code above pm.smaple() only take ~0.5s, so the whole program should not be that slow. Maybe the reality is more complicate than I thought. Many thanks again!