Hi Jura,
Callaway and Sant’Anna allow this explicitly, with covariates.

We, however, only report the overall ATT(g,t) results.

Are you looking to better understand the heterogeneity?

You could subset the data to non-nested groups and see how the effcts vary.

Happy to chat more