Pm.sample gets stuck after init with cores > 1

Have you tried specifiying your priors a bit more? Maybe do pm.Normal(0,5) for predictors/ intercepts?