Poisson

Building Non-Standard Regression Models with JAX

Using JAX’s automatic differentiation to train a Zero-Inflated Generalised Poisson regression model

Modeling correlated count data using bivariate Poisson regression with EM algorithm. Handle overdispersion and correlation in count models.