Baysian hirerchical Linear Regression(Partial Pooling) model using PYMC

the reason I am using different intercepts is that a different location(here US states) have different mortality rates and I want to make a hierarchical model for each location(US State) so for each location I have a separate regression equation

image shows my actual model equation

Screenshot 2024-07-08 at 11.43.28 PM