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.