ZIP

Building Non-Standard Regression Models with JAX

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

• jax
• regression
• count-data
• ZIP
• poisson
• maximum-likelihood