Expose ZeroInflatedLogits and ZeroInflatedProbs from top-level distributions module (#2197)

Author: Qazalbash

numpyro/distributions/__init__.py

@@ -91,7 +91,9 @@
     OrderedLogistic,
     Poisson,
     ZeroInflatedDistribution,
+    ZeroInflatedLogits,
     ZeroInflatedPoisson,
+    ZeroInflatedProbs,
 )
 from numpyro.distributions.distribution import (
     Delta,
@@ -227,7 +229,9 @@
     "Wishart",
     "WishartCholesky",
     "ZeroInflatedDistribution",
+    "ZeroInflatedLogits",
     "ZeroInflatedNegativeBinomial2",
     "ZeroInflatedPoisson",
+    "ZeroInflatedProbs",
     "ZeroSumNormal",
 ]