Distributions
=============

Base Distribution
-----------------

Distribution
^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.distribution.Distribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ExpandedDistribution
^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.distribution.ExpandedDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

FoldedDistribution
^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.distribution.FoldedDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ImproperUniform
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.distribution.ImproperUniform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Independent
^^^^^^^^^^^
.. autoclass:: numpyro.distributions.distribution.Independent
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

MaskedDistribution
^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.distribution.MaskedDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

TransformedDistribution
^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.distribution.TransformedDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Delta
^^^^^
.. autoclass:: numpyro.distributions.distribution.Delta
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Unit
^^^^
.. autoclass:: numpyro.distributions.distribution.Unit
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource


Continuous Distributions
------------------------

AsymmetricLaplace
^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.AsymmetricLaplace
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

AsymmetricLaplaceQuantile
^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.AsymmetricLaplaceQuantile
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Beta
^^^^
.. autoclass:: numpyro.distributions.continuous.Beta
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

BetaProportion
^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.BetaProportion
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

CAR
^^^
.. autoclass:: numpyro.distributions.continuous.CAR
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Cauchy
^^^^^^
.. autoclass:: numpyro.distributions.continuous.Cauchy
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Chi2
^^^^
.. autoclass:: numpyro.distributions.continuous.Chi2
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

CirculantNormal
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.CirculantNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Dagum
^^^^^
.. autoclass:: numpyro.distributions.continuous.Dagum
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Dirichlet
^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.Dirichlet
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

EulerMaruyama
^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.EulerMaruyama
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Exponential
^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.Exponential
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Gamma
^^^^^
.. autoclass:: numpyro.distributions.continuous.Gamma
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

GaussianCopula
^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.copula.GaussianCopula
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

GaussianCopulaBeta
^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.copula.GaussianCopulaBeta
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

GaussianRandomWalk
^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.GaussianRandomWalk
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

GaussianStateSpace
^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.GaussianStateSpace
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Gompertz
^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.Gompertz
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Gumbel
^^^^^^
.. autoclass:: numpyro.distributions.continuous.Gumbel
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

HalfCauchy
^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.HalfCauchy
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

HalfNormal
^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.HalfNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

HurdleGamma
^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.HurdleGamma
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

HurdleLogNormal
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.HurdleLogNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

InverseGamma
^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.InverseGamma
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

InverseWishart
^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.InverseWishart
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

InverseWishartCholesky
^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.InverseWishartCholesky
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Kumaraswamy
^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.Kumaraswamy
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Laplace
^^^^^^^
.. autoclass:: numpyro.distributions.continuous.Laplace
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Levy
^^^^
.. autoclass:: numpyro.distributions.continuous.Levy
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

LKJ
^^^
.. autoclass:: numpyro.distributions.continuous.LKJ
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

LKJCholesky
^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.LKJCholesky
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

LogNormal
^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.LogNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

LogUniform
^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.LogUniform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Logistic
^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.Logistic
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

LowRankMultivariateNormal
^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.LowRankMultivariateNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

MatrixNormal
^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.MatrixNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

MultivariateNormal
^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.MultivariateNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

MultivariateStudentT
^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.MultivariateStudentT
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Normal
^^^^^^
.. autoclass:: numpyro.distributions.continuous.Normal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Pareto
^^^^^^
.. autoclass:: numpyro.distributions.continuous.Pareto
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

RelaxedBernoulli
^^^^^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.continuous.RelaxedBernoulli

RelaxedBernoulliLogits
^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.RelaxedBernoulliLogits
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

SoftLaplace
^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.SoftLaplace
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

StudentT
^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.StudentT
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Uniform
^^^^^^^
.. autoclass:: numpyro.distributions.continuous.Uniform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Weibull
^^^^^^^
.. autoclass:: numpyro.distributions.continuous.Weibull
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Wishart
^^^^^^^
.. autoclass:: numpyro.distributions.continuous.Wishart
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

WishartCholesky
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.WishartCholesky
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ZeroSumNormal
^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.continuous.ZeroSumNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Discrete Distributions
----------------------

Bernoulli
^^^^^^^^^
.. autofunction:: numpyro.distributions.discrete.Bernoulli

BernoulliLogits
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.BernoulliLogits
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

BernoulliProbs
^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.BernoulliProbs
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

BetaBinomial
^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.conjugate.BetaBinomial
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

BetaNegativeBinomial
^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.conjugate.BetaNegativeBinomial
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Binomial
^^^^^^^^^
.. autofunction:: numpyro.distributions.discrete.Binomial

BinomialLogits
^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.BinomialLogits
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

BinomialProbs
^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.BinomialProbs
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Categorical
^^^^^^^^^^^
.. autofunction:: numpyro.distributions.discrete.Categorical

CategoricalLogits
^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.CategoricalLogits
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

CategoricalProbs
^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.CategoricalProbs
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

DirichletMultinomial
^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.conjugate.DirichletMultinomial
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

DiscreteUniform
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.DiscreteUniform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

GammaPoisson
^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.conjugate.GammaPoisson
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Geometric
^^^^^^^^^
.. autofunction:: numpyro.distributions.discrete.Geometric

GeometricLogits
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.GeometricLogits
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

GeometricProbs
^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.GeometricProbs
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

HurdleDistribution
^^^^^^^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.discrete.HurdleDistribution

HurdleLogits
^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.HurdleLogits
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

HurdleNegativeBinomial2
^^^^^^^^^^^^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.conjugate.HurdleNegativeBinomial2

HurdlePoisson
^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.HurdlePoisson
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

HurdleProbs
^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.HurdleProbs
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Multinomial
^^^^^^^^^^^
.. autofunction:: numpyro.distributions.discrete.Multinomial

MultinomialLogits
^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.MultinomialLogits
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

MultinomialProbs
^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.MultinomialProbs
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

OrderedLogistic
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.OrderedLogistic
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

NegativeBinomial
^^^^^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.conjugate.NegativeBinomial

NegativeBinomialLogits
^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.conjugate.NegativeBinomialLogits
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

NegativeBinomialProbs
^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.conjugate.NegativeBinomialProbs
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

NegativeBinomial2
^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.conjugate.NegativeBinomial2
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Poisson
^^^^^^^
.. autoclass:: numpyro.distributions.discrete.Poisson
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ZeroInflatedDistribution
^^^^^^^^^^^^^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.discrete.ZeroInflatedDistribution

ZeroInflatedPoisson
^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.discrete.ZeroInflatedPoisson
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ZeroInflatedNegativeBinomial2
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.conjugate.ZeroInflatedNegativeBinomial2


Mixture Distributions
---------------------

Mixture
^^^^^^^
.. autofunction:: numpyro.distributions.mixtures.Mixture

MixtureSameFamily
^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.mixtures.MixtureSameFamily
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

MixtureGeneral
^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.mixtures.MixtureGeneral
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Directional Distributions
-------------------------

ProjectedNormal
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.directional.ProjectedNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

SineBivariateVonMises
^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.directional.SineBivariateVonMises
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

SineSkewed
^^^^^^^^^^
.. autoclass:: numpyro.distributions.directional.SineSkewed
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

VonMises
^^^^^^^^
.. autoclass:: numpyro.distributions.directional.VonMises
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource


Truncated Distributions
-----------------------

DoublyTruncatedPowerLaw
^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.truncated.DoublyTruncatedPowerLaw
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

LeftTruncatedDistribution
^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.truncated.LeftTruncatedDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

LowerTruncatedPowerLaw
^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.truncated.LowerTruncatedPowerLaw
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

RightTruncatedDistribution
^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.truncated.RightTruncatedDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

TruncatedCauchy
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.truncated.TruncatedCauchy
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

TruncatedDistribution
^^^^^^^^^^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.truncated.TruncatedDistribution

TruncatedNormal
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.truncated.TruncatedNormal
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

TruncatedPolyaGamma
^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.truncated.TruncatedPolyaGamma
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

TwoSidedTruncatedDistribution
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.truncated.TwoSidedTruncatedDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

Censored Distributions
-----------------------

LeftCensoredDistribution
^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.censored.LeftCensoredDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

RightCensoredDistribution
^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.censored.RightCensoredDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

IntervalCensoredDistribution
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.censored.IntervalCensoredDistribution
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

TensorFlow Distributions
------------------------

Thin wrappers around TensorFlow Probability (TFP) distributions. For details on the TFP distribution interface,
see `its Distribution docs <https://www.tensorflow.org/probability/api_docs/python/tfp/substrates/jax/distributions/Distribution>`_.

.. automodule:: numpyro.contrib.tfp.distributions


Constraints
-----------

Constraint
^^^^^^^^^^
.. autoclass:: numpyro.distributions.constraints.Constraint
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

boolean
^^^^^^^
.. autodata:: numpyro.distributions.constraints.boolean

circular
^^^^^^^^
.. autodata:: numpyro.distributions.constraints.circular

corr_cholesky
^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.corr_cholesky

corr_matrix
^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.corr_matrix

dependent
^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.dependent

greater_than
^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.constraints.greater_than

integer_interval
^^^^^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.constraints.integer_interval

integer_greater_than
^^^^^^^^^^^^^^^^^^^^
.. autofunction:: numpyro.distributions.constraints.integer_greater_than

interval
^^^^^^^^
.. autofunction:: numpyro.distributions.constraints.interval

l1_ball
^^^^^^^
.. autofunction:: numpyro.distributions.constraints.l1_ball

less_than
^^^^^^^^^
.. autofunction:: numpyro.distributions.constraints.less_than

lower_cholesky
^^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.lower_cholesky

multinomial
^^^^^^^^^^^
.. autofunction:: numpyro.distributions.constraints.multinomial

nonnegative_integer
^^^^^^^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.nonnegative_integer

ordered_vector
^^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.ordered_vector

positive
^^^^^^^^
.. autodata:: numpyro.distributions.constraints.positive

positive_definite
^^^^^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.positive_definite

positive_integer
^^^^^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.positive_integer

positive_ordered_vector
^^^^^^^^^^^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.positive_ordered_vector

real
^^^^
.. autodata:: numpyro.distributions.constraints.real

real_vector
^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.real_vector

scaled_unit_lower_cholesky
^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.scaled_unit_lower_cholesky

softplus_positive
^^^^^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.softplus_positive

softplus_lower_cholesky
^^^^^^^^^^^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.softplus_lower_cholesky

simplex
^^^^^^^
.. autodata:: numpyro.distributions.constraints.simplex

sphere
^^^^^^
.. autodata:: numpyro.distributions.constraints.sphere

unit_interval
^^^^^^^^^^^^^
.. autodata:: numpyro.distributions.constraints.unit_interval

zero_sum
^^^^^^^^
.. autodata:: numpyro.distributions.constraints.zero_sum

Transforms
----------

biject_to
^^^^^^^^^
.. autofunction:: numpyro.distributions.transforms.biject_to

Transform
^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.Transform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

AbsTransform
^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.AbsTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

AffineTransform
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.AffineTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

CholeskyTransform
^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.CholeskyTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ComplexTransform
^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.ComplexTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ComposeTransform
^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.ComposeTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

CorrCholeskyTransform
^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.CorrCholeskyTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

CorrMatrixCholeskyTransform
^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.CorrMatrixCholeskyTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ExpTransform
^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.ExpTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

IdentityTransform
^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.IdentityTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

L1BallTransform
^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.L1BallTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

LowerCholeskyAffine
^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.LowerCholeskyAffine
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

LowerCholeskyTransform
^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.LowerCholeskyTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

OrderedTransform
^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.OrderedTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

PackRealFastFourierCoefficientsTransform
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.PackRealFastFourierCoefficientsTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

PermuteTransform
^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.PermuteTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

PowerTransform
^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.PowerTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

RealFastFourierTransform
^^^^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: numpyro.distributions.transforms.RealFastFourierTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

RecursiveLinearTransform
^^^^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: numpyro.distributions.transforms.RecursiveLinearTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ScaledUnitLowerCholeskyTransform
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.ScaledUnitLowerCholeskyTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

SigmoidTransform
^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.SigmoidTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

SimplexToOrderedTransform
^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.SimplexToOrderedTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

SoftplusLowerCholeskyTransform
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.SoftplusLowerCholeskyTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

SoftplusTransform
^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.SoftplusTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

StickBreakingTransform
^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.transforms.StickBreakingTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

ZeroSumTransform
^^^^^^^^^^^^^^^^

.. autoclass:: numpyro.distributions.transforms.ZeroSumTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource


Flows
-----

InverseAutoregressiveTransform
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.flows.InverseAutoregressiveTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource

BlockNeuralAutoregressiveTransform
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: numpyro.distributions.flows.BlockNeuralAutoregressiveTransform
    :members:
    :undoc-members:
    :show-inheritance:
    :member-order: bysource


Utilities
---------

log1mexp
^^^^^^^^
.. autofunction:: numpyro.distributions.util.log1mexp

logdiffexp
^^^^^^^^^^
.. autofunction:: numpyro.distributions.util.logdiffexp
