In my line of work I've been using numpyro to model physical phenomena. There are multiple unbalanced categorical variables, so partial pooling helps a lot to

In my line of work I've been using numpyro to model physical phenomena. There are multiple unbalanced categorical variables, so partial pooling helps a lot to infer the target in regions where the data is sparse.