NumPyro
0.7.0
Getting Started with NumPyro
API and Developer Reference
Pyro Primitives
Distributions
Inference
Effect Handlers
Contributed Code
Change Log
Introductory Tutorials
Bayesian Regression Using NumPyro
Bayesian Hierarchical Linear Regression
Example: Baseball Batting Average
Example: Variational Autoencoder
Example: Neal’s Funnel
Example: Stochastic Volatility
Automatic rendering of NumPyro models
Discrete Latent Variables
Example: Bayesian Models of Annotation
Example: Enumerate Hidden Markov Model
Example: CJS Capture-Recapture Model for Ecological Data
Example: Nested Sampling for Gaussian Shells
Bayesian Imputation for Missing Values in Discrete Covariates
Example: ProdLDA with Flax and Haiku
Applications
Time Series Forecasting
Ordinal Regression
Bayesian Imputation
Example: Gaussian Process
Example: Bayesian Neural Network
Example: Sparse Regression
Example: Proportion Test
Example: Generalized Linear Mixed Models
Example: Hamiltonian Monte Carlo with Energy Conserving Subsampling
Example: Hidden Markov Model
Example: Predator-Prey Model
Example: Neural Transport
Example: MCMC Methods for Tall Data
Example: Thompson sampling for Bayesian Optimization with GPs
NumPyro
Docs
»
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
J
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
Z
_
__call__() (LocScaleReparam method)
(NeuTraReparam method)
(ProjectedNormalReparam method)
(TransformReparam method)
A
AbsTransform (class in numpyro.distributions.transforms)
Adagrad (class in numpyro.optim)
Adam (class in numpyro.optim)
AffineTransform (class in numpyro.distributions.transforms)
arg_constraints (BernoulliLogits attribute)
(BernoulliProbs attribute)
(Beta attribute)
(BetaBinomial attribute)
(BetaProportion attribute)
(BinomialLogits attribute)
(BinomialProbs attribute)
(CategoricalLogits attribute)
(CategoricalProbs attribute)
(Cauchy attribute)
(Chi2 attribute)
(Delta attribute)
(Dirichlet attribute)
(DirichletMultinomial attribute)
(Distribution attribute)
(ExpandedDistribution attribute)
(Exponential attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(GeometricLogits attribute)
(GeometricProbs attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(ImproperUniform attribute)
(Independent attribute)
(InverseGamma attribute)
(LKJ attribute)
(LKJCholesky attribute)
(Laplace attribute)
(LeftTruncatedDistribution attribute)
(LogNormal attribute)
(Logistic attribute)
(LowRankMultivariateNormal attribute)
(MaskedDistribution attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(NegativeBinomial2 attribute)
(NegativeBinomialLogits attribute)
(NegativeBinomialProbs attribute)
(Normal attribute)
(OrderedLogistic attribute)
(Pareto attribute)
(Poisson attribute)
(ProjectedNormal attribute)
(RightTruncatedDistribution attribute)
(SoftLaplace attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(TruncatedPolyaGamma attribute)
(TwoSidedTruncatedDistribution attribute)
(Uniform attribute)
(Unit attribute)
(VonMises attribute)
(Weibull attribute)
(ZeroInflatedPoisson attribute)
AutoBNAFNormal (class in numpyro.infer.autoguide)
AutoContinuous (class in numpyro.infer.autoguide)
autocorrelation() (in module numpyro.diagnostics)
autocovariance() (in module numpyro.diagnostics)
AutoDelta (class in numpyro.infer.autoguide)
AutoDiagonalNormal (class in numpyro.infer.autoguide)
AutoGuide (class in numpyro.infer.autoguide)
AutoIAFNormal (class in numpyro.infer.autoguide)
AutoLaplaceApproximation (class in numpyro.infer.autoguide)
AutoLowRankMultivariateNormal (class in numpyro.infer.autoguide)
AutoMultivariateNormal (class in numpyro.infer.autoguide)
AutoNormal (class in numpyro.infer.autoguide)
Autoregressive (class in numpyro.contrib.tfp.distributions)
B
BarkerMH (class in numpyro.infer.barker)
BarkerMHState (in module numpyro.infer.barker)
batch_shape (Distribution attribute)
BatchBroadcast (class in numpyro.contrib.tfp.distributions)
BatchConcat (class in numpyro.contrib.tfp.distributions)
BatchReshape (class in numpyro.contrib.tfp.distributions)
Bates (class in numpyro.contrib.tfp.distributions)
Bernoulli (class in numpyro.contrib.tfp.distributions)
Bernoulli() (in module numpyro.distributions.discrete)
BernoulliLogits (class in numpyro.distributions.discrete)
BernoulliProbs (class in numpyro.distributions.discrete)
Beta (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
BetaBinomial (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.conjugate)
BetaProportion (class in numpyro.distributions.continuous)
BetaQuotient (class in numpyro.contrib.tfp.distributions)
biject_to() (in module numpyro.distributions.transforms)
BijectorConstraint (class in numpyro.contrib.tfp.distributions)
BijectorTransform (class in numpyro.contrib.tfp.distributions)
Binomial (class in numpyro.contrib.tfp.distributions)
Binomial() (in module numpyro.distributions.discrete)
BinomialLogits (class in numpyro.distributions.discrete)
BinomialProbs (class in numpyro.distributions.discrete)
block (class in numpyro.handlers)
BlockNeuralAutoregressiveTransform (class in numpyro.distributions.flows)
Blockwise (class in numpyro.contrib.tfp.distributions)
boolean (in module numpyro.distributions.constraints)
C
call_with_intermediates() (BlockNeuralAutoregressiveTransform method)
(ComposeTransform method)
(InverseAutoregressiveTransform method)
(Transform method)
Categorical (class in numpyro.contrib.tfp.distributions)
Categorical() (in module numpyro.distributions.discrete)
CategoricalLogits (class in numpyro.distributions.discrete)
CategoricalProbs (class in numpyro.distributions.discrete)
Cauchy (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
cdf() (Beta method)
(Cauchy method)
(Distribution method)
(Exponential method)
(GammaPoisson method)
(Gumbel method)
(HalfCauchy method)
(HalfNormal method)
(Laplace method)
(Logistic method)
(Normal method)
(Pareto method)
(Poisson method)
(SoftLaplace method)
(StudentT method)
(Uniform method)
(Weibull method)
check() (Constraint method)
Chi (class in numpyro.contrib.tfp.distributions)
Chi2 (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
CholeskyLKJ (class in numpyro.contrib.tfp.distributions)
CholeskyTransform (class in numpyro.distributions.transforms)
ClippedAdam (class in numpyro.optim)
codomain (AbsTransform attribute)
(AffineTransform attribute)
(BlockNeuralAutoregressiveTransform attribute)
(CholeskyTransform attribute)
(ComposeTransform attribute)
(CorrCholeskyTransform attribute)
(CorrMatrixCholeskyTransform attribute)
(ExpTransform attribute)
(InvCholeskyTransform attribute)
(InverseAutoregressiveTransform attribute)
(LowerCholeskyAffine attribute)
(LowerCholeskyTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(PowerTransform attribute)
(SigmoidTransform attribute)
(SoftplusLowerCholeskyTransform attribute)
(SoftplusTransform attribute)
(StickBreakingTransform attribute)
(Transform attribute)
collapse (class in numpyro.handlers)
ComposeTransform (class in numpyro.distributions.transforms)
cond() (in module numpyro.contrib.control_flow)
condition (class in numpyro.handlers)
config_enumerate() (in module numpyro.contrib.funsor.infer_util)
consensus() (in module numpyro.infer.hmc_util)
constrain_fn() (in module numpyro.infer.util)
Constraint (class in numpyro.distributions.constraints)
ContinuousBernoulli (class in numpyro.contrib.tfp.distributions)
corr_cholesky (in module numpyro.distributions.constraints)
corr_matrix (in module numpyro.distributions.constraints)
CorrCholeskyTransform (class in numpyro.distributions.transforms)
CorrMatrixCholeskyTransform (class in numpyro.distributions.transforms)
covariance_matrix (LowRankMultivariateNormal attribute)
(MultivariateNormal attribute)
D
default_fields (HMC attribute)
(MCMCKernel attribute)
(SA attribute)
Delta (class in numpyro.distributions.distribution)
dependent (in module numpyro.distributions.constraints)
DeterminantalPointProcess (class in numpyro.contrib.tfp.distributions)
Deterministic (class in numpyro.contrib.tfp.distributions)
deterministic() (in module numpyro.primitives)
diagnostics() (NestedSampler method)
Dirichlet (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
DirichletMultinomial (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.conjugate)
DiscreteHMCGibbs (class in numpyro.infer.hmc_gibbs)
Distribution (class in numpyro.distributions.distribution)
do (class in numpyro.handlers)
domain (AbsTransform attribute)
(BlockNeuralAutoregressiveTransform attribute)
(CholeskyTransform attribute)
(ComposeTransform attribute)
(CorrCholeskyTransform attribute)
(CorrMatrixCholeskyTransform attribute)
(InverseAutoregressiveTransform attribute)
(LowerCholeskyAffine attribute)
(LowerCholeskyTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(PowerTransform attribute)
(SoftplusLowerCholeskyTransform attribute)
(SoftplusTransform attribute)
(StickBreakingTransform attribute)
(Transform attribute)
DoublesidedMaxwell (class in numpyro.contrib.tfp.distributions)
E
effective_sample_size() (in module numpyro.diagnostics)
ELBO (class in numpyro.infer.elbo)
Empirical (class in numpyro.contrib.tfp.distributions)
enable_validation() (in module numpyro.distributions.distribution)
enable_x64() (in module numpyro.util)
entropy() (LowRankMultivariateNormal method)
enum (class in numpyro.contrib.funsor.enum_messenger)
enumerate_support() (BernoulliLogits method)
(BernoulliProbs method)
(BetaBinomial method)
(BinomialLogits method)
(BinomialProbs method)
(CategoricalLogits method)
(CategoricalProbs method)
(Distribution method)
(ExpandedDistribution method)
(MaskedDistribution method)
eval_and_stable_update() (Adagrad method)
(Adam method)
(ClippedAdam method)
(Minimize method)
(Momentum method)
(RMSProp method)
(RMSPropMomentum method)
(SGD method)
(SM3 method)
eval_and_update() (Adagrad method)
(Adam method)
(ClippedAdam method)
(Minimize method)
(Momentum method)
(RMSProp method)
(RMSPropMomentum method)
(SGD method)
(SM3 method)
evaluate() (SVI method)
event_dim (Constraint attribute)
(Distribution attribute)
(Transform attribute)
event_shape (Distribution attribute)
expand() (Distribution method)
(Independent method)
expand_by() (Distribution method)
ExpandedDistribution (class in numpyro.distributions.distribution)
ExpGamma (class in numpyro.contrib.tfp.distributions)
ExpInverseGamma (class in numpyro.contrib.tfp.distributions)
Exponential (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
ExponentiallyModifiedGaussian (class in numpyro.contrib.tfp.distributions)
ExpRelaxedOneHotCategorical (class in numpyro.contrib.tfp.distributions)
ExpTransform (class in numpyro.distributions.transforms)
F
factor() (in module numpyro.primitives)
feasible_like() (Constraint method)
find_valid_initial_params() (in module numpyro.infer.util)
FiniteDiscrete (class in numpyro.contrib.tfp.distributions)
flax_module() (in module numpyro.contrib.module)
FoldedDistribution (class in numpyro.distributions.distribution)
fori_collect() (in module numpyro.util)
forward_shape() (AffineTransform method)
(ComposeTransform method)
(CorrCholeskyTransform method)
(LowerCholeskyAffine method)
(LowerCholeskyTransform method)
(PowerTransform method)
(SoftplusLowerCholeskyTransform method)
(StickBreakingTransform method)
(Transform method)
G
Gamma (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
GammaGamma (class in numpyro.contrib.tfp.distributions)
GammaPoisson (class in numpyro.distributions.conjugate)
GaussianProcess (class in numpyro.contrib.tfp.distributions)
GaussianProcessRegressionModel (class in numpyro.contrib.tfp.distributions)
GaussianRandomWalk (class in numpyro.distributions.continuous)
gelman_rubin() (in module numpyro.diagnostics)
GeneralizedExtremeValue (class in numpyro.contrib.tfp.distributions)
GeneralizedNormal (class in numpyro.contrib.tfp.distributions)
GeneralizedPareto (class in numpyro.contrib.tfp.distributions)
Geometric (class in numpyro.contrib.tfp.distributions)
Geometric() (in module numpyro.distributions.discrete)
GeometricLogits (class in numpyro.distributions.discrete)
GeometricProbs (class in numpyro.distributions.discrete)
get_base_dist() (AutoBNAFNormal method)
(AutoContinuous method)
(AutoDiagonalNormal method)
(AutoIAFNormal method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
get_diagnostics_str() (BarkerMH method)
(HMC method)
(HMCGibbs method)
(MCMCKernel method)
(SA method)
get_extra_fields() (MCMC method)
get_mask() (in module numpyro.primitives)
get_model_relations() (in module numpyro.contrib.render)
get_params() (Adagrad method)
(Adam method)
(ClippedAdam method)
(Minimize method)
(Momentum method)
(RMSProp method)
(RMSPropMomentum method)
(SGD method)
(SM3 method)
(SVI method)
get_posterior() (AutoContinuous method)
(AutoDiagonalNormal method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
get_samples() (MCMC method)
(NestedSampler method)
get_trace() (trace method)
get_transform() (AutoContinuous method)
(AutoDiagonalNormal method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
get_weighted_samples() (NestedSampler method)
greater_than() (in module numpyro.distributions.constraints)
Gumbel (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
H
haiku_module() (in module numpyro.contrib.module)
HalfCauchy (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
HalfNormal (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
HalfStudentT (class in numpyro.contrib.tfp.distributions)
HamiltonianMonteCarlo (class in numpyro.contrib.tfp.mcmc)
has_enumerate_support (BernoulliLogits attribute)
(BernoulliProbs attribute)
(BetaBinomial attribute)
(BinomialLogits attribute)
(BinomialProbs attribute)
(CategoricalLogits attribute)
(CategoricalProbs attribute)
(Distribution attribute)
(ExpandedDistribution attribute)
(Independent attribute)
(MaskedDistribution attribute)
has_rsample (Distribution attribute)
(ExpandedDistribution attribute)
(Independent attribute)
(MaskedDistribution attribute)
(TransformedDistribution attribute)
HiddenMarkovModel (class in numpyro.contrib.tfp.distributions)
HMC (class in numpyro.infer.hmc)
hmc() (in module numpyro.infer.hmc)
HMCECS (class in numpyro.infer.hmc_gibbs)
HMCGibbs (class in numpyro.infer.hmc_gibbs)
HMCGibbsState (in module numpyro.infer.hmc_gibbs)
HMCState (in module numpyro.infer.hmc)
Horseshoe (class in numpyro.contrib.tfp.distributions)
hpdi() (in module numpyro.diagnostics)
I
icdf() (Cauchy method)
(Distribution method)
(Exponential method)
(Gumbel method)
(HalfCauchy method)
(HalfNormal method)
(Laplace method)
(Logistic method)
(Normal method)
(Pareto method)
(SoftLaplace method)
(StudentT method)
(Uniform method)
IdentityTransform (class in numpyro.distributions.transforms)
ImproperUniform (class in numpyro.distributions.distribution)
Independent (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.distribution)
infer_config (class in numpyro.contrib.funsor.enum_messenger)
(class in numpyro.handlers)
infer_discrete() (in module numpyro.contrib.funsor.discrete)
infer_shapes() (Dirichlet static method)
(DirichletMultinomial static method)
(LowRankMultivariateNormal static method)
(MultinomialLogits static method)
(MultinomialProbs static method)
(MultivariateNormal static method)
(OrderedLogistic static method)
(ProjectedNormal static method)
(Uniform static method)
(numpyro.distributions.distribution.Distribution class method)
init() (Adagrad method)
(Adam method)
(BarkerMH method)
(ClippedAdam method)
(DiscreteHMCGibbs method)
(HMC method)
(HMCECS method)
(HMCGibbs method)
(MCMCKernel method)
(Minimize method)
(MixedHMC method)
(Momentum method)
(RMSProp method)
(RMSPropMomentum method)
(SA method)
(SGD method)
(SM3 method)
(SVI method)
init_kernel() (in module numpyro.infer.hmc.hmc)
init_to_feasible() (in module numpyro.infer.initialization)
init_to_median() (in module numpyro.infer.initialization)
init_to_sample() (in module numpyro.infer.initialization)
init_to_uniform() (in module numpyro.infer.initialization)
init_to_value() (in module numpyro.infer.initialization)
initialize_model() (in module numpyro.infer.util)
integer_greater_than() (in module numpyro.distributions.constraints)
integer_interval() (in module numpyro.distributions.constraints)
interval() (in module numpyro.distributions.constraints)
inv (Transform attribute)
InvCholeskyTransform (class in numpyro.distributions.transforms)
inverse_shape() (AffineTransform method)
(ComposeTransform method)
(CorrCholeskyTransform method)
(LowerCholeskyAffine method)
(LowerCholeskyTransform method)
(PowerTransform method)
(SoftplusLowerCholeskyTransform method)
(StickBreakingTransform method)
(Transform method)
InverseAutoregressiveTransform (class in numpyro.distributions.flows)
InverseGamma (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
InverseGaussian (class in numpyro.contrib.tfp.distributions)
is_discrete (Constraint attribute)
(Distribution attribute)
J
JohnsonSU (class in numpyro.contrib.tfp.distributions)
JointDistribution (class in numpyro.contrib.tfp.distributions)
JointDistributionCoroutine (class in numpyro.contrib.tfp.distributions)
JointDistributionCoroutineAutoBatched (class in numpyro.contrib.tfp.distributions)
JointDistributionNamed (class in numpyro.contrib.tfp.distributions)
JointDistributionNamedAutoBatched (class in numpyro.contrib.tfp.distributions)
JointDistributionSequential (class in numpyro.contrib.tfp.distributions)
JointDistributionSequentialAutoBatched (class in numpyro.contrib.tfp.distributions)
K
Kumaraswamy (class in numpyro.contrib.tfp.distributions)
L
LambertWDistribution (class in numpyro.contrib.tfp.distributions)
LambertWNormal (class in numpyro.contrib.tfp.distributions)
Laplace (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
last_state (MCMC attribute)
LeftTruncatedDistribution (class in numpyro.distributions.truncated)
less_than() (in module numpyro.distributions.constraints)
lift (class in numpyro.handlers)
LinearGaussianStateSpaceModel (class in numpyro.contrib.tfp.distributions)
LKJ (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
LKJCholesky (class in numpyro.distributions.continuous)
LocScaleReparam (class in numpyro.infer.reparam)
log_abs_det_jacobian() (AffineTransform method)
(BlockNeuralAutoregressiveTransform method)
(CholeskyTransform method)
(ComposeTransform method)
(CorrCholeskyTransform method)
(CorrMatrixCholeskyTransform method)
(ExpTransform method)
(IdentityTransform method)
(InvCholeskyTransform method)
(InverseAutoregressiveTransform method)
(LowerCholeskyAffine method)
(LowerCholeskyTransform method)
(OrderedTransform method)
(PermuteTransform method)
(PowerTransform method)
(SigmoidTransform method)
(SoftplusLowerCholeskyTransform method)
(SoftplusTransform method)
(StickBreakingTransform method)
(Transform method)
log_density() (in module numpyro.contrib.funsor.infer_util)
(in module numpyro.infer.util)
log_likelihood() (in module numpyro.infer.util)
log_prob() (BernoulliLogits method)
(BernoulliProbs method)
(Beta method)
(BetaBinomial method)
(BinomialLogits method)
(BinomialProbs method)
(CategoricalLogits method)
(CategoricalProbs method)
(Cauchy method)
(Delta method)
(Dirichlet method)
(DirichletMultinomial method)
(Distribution method)
(ExpandedDistribution method)
(Exponential method)
(FoldedDistribution method)
(Gamma method)
(GammaPoisson method)
(GaussianRandomWalk method)
(GeometricLogits method)
(GeometricProbs method)
(Gumbel method)
(HalfCauchy method)
(HalfNormal method)
(ImproperUniform method)
(Independent method)
(LKJCholesky method)
(Laplace method)
(LeftTruncatedDistribution method)
(Logistic method)
(LowRankMultivariateNormal method)
(MaskedDistribution method)
(MultinomialLogits method)
(MultinomialProbs method)
(MultivariateNormal method)
(NegativeBinomialLogits method)
(Normal method)
(Poisson method)
(ProjectedNormal method)
(RightTruncatedDistribution method)
(SoftLaplace method)
(StudentT method)
(TransformedDistribution method)
(TruncatedPolyaGamma method)
(TwoSidedTruncatedDistribution method)
(Uniform method)
(Unit method)
(VonMises method)
(Weibull method)
Logistic (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
LogitNormal (class in numpyro.contrib.tfp.distributions)
logits (BernoulliProbs attribute)
(BinomialProbs attribute)
(CategoricalProbs attribute)
(GeometricProbs attribute)
(MultinomialProbs attribute)
LogLogistic (class in numpyro.contrib.tfp.distributions)
LogNormal (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
loss() (ELBO method)
(RenyiELBO method)
(TraceGraph_ELBO method)
loss_with_mutable_state() (ELBO method)
(TraceMeanField_ELBO method)
(Trace_ELBO method)
lower_cholesky (in module numpyro.distributions.constraints)
LowerCholeskyAffine (class in numpyro.distributions.transforms)
LowerCholeskyTransform (class in numpyro.distributions.transforms)
LowRankMultivariateNormal (class in numpyro.distributions.continuous)
M
markov() (in module numpyro.contrib.funsor.enum_messenger)
mask (class in numpyro.handlers)
mask() (Distribution method)
Masked (class in numpyro.contrib.tfp.distributions)
MaskedDistribution (class in numpyro.distributions.distribution)
MatrixNormalLinearOperator (class in numpyro.contrib.tfp.distributions)
MatrixTLinearOperator (class in numpyro.contrib.tfp.distributions)
MCMC (class in numpyro.infer.mcmc)
MCMCKernel (class in numpyro.infer.mcmc)
mean (BernoulliLogits attribute)
(BernoulliProbs attribute)
(Beta attribute)
(BetaBinomial attribute)
(BinomialLogits attribute)
(BinomialProbs attribute)
(CategoricalLogits attribute)
(CategoricalProbs attribute)
(Cauchy attribute)
(Delta attribute)
(Dirichlet attribute)
(DirichletMultinomial attribute)
(Distribution attribute)
(ExpandedDistribution attribute)
(Exponential attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(GeometricLogits attribute)
(GeometricProbs attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(Independent attribute)
(InverseGamma attribute)
(LKJ attribute)
(Laplace attribute)
(LogNormal attribute)
(Logistic attribute)
(LowRankMultivariateNormal attribute)
(MaskedDistribution attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(Poisson attribute)
(ProjectedNormal attribute)
(SoftLaplace attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
(VonMises attribute)
(Weibull attribute)
median() (AutoDelta method)
(AutoDiagonalNormal method)
(AutoGuide method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
(AutoNormal method)
MetropolisAdjustedLangevinAlgorithm (class in numpyro.contrib.tfp.mcmc)
Minimize (class in numpyro.optim)
MixedHMC (class in numpyro.infer.mixed_hmc)
MixtureSameFamily (class in numpyro.contrib.tfp.distributions)
mode (ProjectedNormal attribute)
model (BarkerMH attribute)
(HMC attribute)
(HMCGibbs attribute)
module() (in module numpyro.primitives)
Momentum (class in numpyro.optim)
Moyal (class in numpyro.contrib.tfp.distributions)
Multinomial (class in numpyro.contrib.tfp.distributions)
multinomial() (in module numpyro.distributions.constraints)
Multinomial() (in module numpyro.distributions.discrete)
MultinomialLogits (class in numpyro.distributions.discrete)
MultinomialProbs (class in numpyro.distributions.discrete)
MultivariateNormal (class in numpyro.distributions.continuous)
MultivariateNormalDiag (class in numpyro.contrib.tfp.distributions)
MultivariateNormalDiagPlusLowRank (class in numpyro.contrib.tfp.distributions)
MultivariateNormalDiagPlusLowRankCovariance (class in numpyro.contrib.tfp.distributions)
MultivariateNormalFullCovariance (class in numpyro.contrib.tfp.distributions)
MultivariateNormalLinearOperator (class in numpyro.contrib.tfp.distributions)
MultivariateNormalTriL (class in numpyro.contrib.tfp.distributions)
MultivariateStudentTLinearOperator (class in numpyro.contrib.tfp.distributions)
N
NegativeBinomial (class in numpyro.contrib.tfp.distributions)
NegativeBinomial() (in module numpyro.distributions.conjugate)
NegativeBinomial2 (class in numpyro.distributions.conjugate)
NegativeBinomialLogits (class in numpyro.distributions.conjugate)
NegativeBinomialProbs (class in numpyro.distributions.conjugate)
NestedSampler (class in numpyro.contrib.nested_sampling)
NeuTraReparam (class in numpyro.infer.reparam)
nonnegative_integer (in module numpyro.distributions.constraints)
Normal (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
NormalInverseGaussian (class in numpyro.contrib.tfp.distributions)
NoUTurnSampler (class in numpyro.contrib.tfp.mcmc)
num_gamma_variates (TruncatedPolyaGamma attribute)
num_log_prob_terms (TruncatedPolyaGamma attribute)
numpyro.contrib.funsor (module)
numpyro.contrib.indexing (module)
numpyro.contrib.tfp.distributions (module)
numpyro.contrib.tfp.mcmc (module)
numpyro.diagnostics (module)
numpyro.handlers (module)
numpyro.infer.autoguide (module)
numpyro.infer.reparam (module)
numpyro.infer.util (module)
numpyro.optim (module)
numpyro.primitives (module)
numpyro.util (module)
NUTS (class in numpyro.infer.hmc)
O
OneHotCategorical (class in numpyro.contrib.tfp.distributions)
optax_to_numpyro() (in module numpyro.contrib.optim)
ordered_vector (in module numpyro.distributions.constraints)
OrderedLogistic (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.discrete)
OrderedTransform (class in numpyro.distributions.transforms)
P
param() (in module numpyro.primitives)
parametric() (in module numpyro.infer.hmc_util)
parametric_draws() (in module numpyro.infer.hmc_util)
Pareto (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
PermuteTransform (class in numpyro.distributions.transforms)
PERT (class in numpyro.contrib.tfp.distributions)
PlackettLuce (class in numpyro.contrib.tfp.distributions)
plate (class in numpyro.contrib.funsor.enum_messenger)
(class in numpyro.primitives)
plate_stack() (in module numpyro.primitives)
plate_to_enum_plate() (in module numpyro.contrib.funsor.infer_util)
Poisson (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.discrete)
PoissonLogNormalQuadratureCompound (class in numpyro.contrib.tfp.distributions)
positive (in module numpyro.distributions.constraints)
positive_definite (in module numpyro.distributions.constraints)
positive_integer (in module numpyro.distributions.constraints)
positive_ordered_vector (in module numpyro.distributions.constraints)
post_warmup_state (MCMC attribute)
postprocess_fn() (BarkerMH method)
(HMC method)
(HMCECS method)
(HMCGibbs method)
(MCMCKernel method)
(SA method)
postprocess_message() (plate method)
(trace method)
,
[1]
potential_energy() (in module numpyro.infer.util)
PowerSpherical (class in numpyro.contrib.tfp.distributions)
PowerTransform (class in numpyro.distributions.transforms)
precision_matrix (LowRankMultivariateNormal attribute)
(MultivariateNormal attribute)
Predictive (class in numpyro.infer.util)
print_summary() (in module numpyro.diagnostics)
(MCMC method)
(NestedSampler method)
prng_key() (in module numpyro.primitives)
PRNGIdentity (class in numpyro.distributions.discrete)
ProbitBernoulli (class in numpyro.contrib.tfp.distributions)
probs (BernoulliLogits attribute)
(BinomialLogits attribute)
(CategoricalLogits attribute)
(GeometricLogits attribute)
(MultinomialLogits attribute)
process_message() (block method)
(collapse method)
(condition method)
(do method)
(enum method)
(infer_config method)
,
[1]
(lift method)
(mask method)
(plate method)
(reparam method)
(replay method)
(scale method)
(scope method)
(seed method)
(substitute method)
ProjectedNormal (class in numpyro.distributions.directional)
ProjectedNormalReparam (class in numpyro.infer.reparam)
Q
quantiles() (AutoDiagonalNormal method)
(AutoGuide method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
(AutoNormal method)
QuantizedDistribution (class in numpyro.contrib.tfp.distributions)
R
random_flax_module() (in module numpyro.contrib.module)
random_haiku_module() (in module numpyro.contrib.module)
RandomWalkMetropolis (class in numpyro.contrib.tfp.mcmc)
real (in module numpyro.distributions.constraints)
real_vector (in module numpyro.distributions.constraints)
RelaxedBernoulli (class in numpyro.contrib.tfp.distributions)
RelaxedOneHotCategorical (class in numpyro.contrib.tfp.distributions)
render_model() (in module numpyro.contrib.render)
RenyiELBO (class in numpyro.infer.elbo)
reparam (class in numpyro.handlers)
Reparam (class in numpyro.infer.reparam)
reparam() (NeuTraReparam method)
reparameterized_params (Independent attribute)
reparametrized_params (Beta attribute)
(BetaProportion attribute)
(Cauchy attribute)
(Chi2 attribute)
(Delta attribute)
(Dirichlet attribute)
(Distribution attribute)
(Exponential attribute)
(Gamma attribute)
(GaussianRandomWalk attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(InverseGamma attribute)
(LKJ attribute)
(LKJCholesky attribute)
(Laplace attribute)
(LeftTruncatedDistribution attribute)
(LogNormal attribute)
(Logistic attribute)
(LowRankMultivariateNormal attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(ProjectedNormal attribute)
(RightTruncatedDistribution attribute)
(SoftLaplace attribute)
(StudentT attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(TwoSidedTruncatedDistribution attribute)
(Uniform attribute)
(VonMises attribute)
(Weibull attribute)
replay (class in numpyro.handlers)
ReplicaExchangeMC (class in numpyro.contrib.tfp.mcmc)
RightTruncatedDistribution (class in numpyro.distributions.truncated)
RMSProp (class in numpyro.optim)
RMSPropMomentum (class in numpyro.optim)
rsample() (Distribution method)
(ExpandedDistribution method)
(Independent method)
(MaskedDistribution method)
(TransformedDistribution method)
run() (MCMC method)
(NestedSampler method)
(SVI method)
S
SA (class in numpyro.infer.sa)
Sample (class in numpyro.contrib.tfp.distributions)
sample() (BarkerMH method)
(BernoulliLogits method)
(BernoulliProbs method)
(Beta method)
(BetaBinomial method)
(BinomialLogits method)
(BinomialProbs method)
(CategoricalLogits method)
(CategoricalProbs method)
(Cauchy method)
(Delta method)
(Dirichlet method)
(DirichletMultinomial method)
(DiscreteHMCGibbs method)
(Distribution method)
(ExpandedDistribution method)
(Exponential method)
(Gamma method)
(GammaPoisson method)
(GaussianRandomWalk method)
(GeometricLogits method)
(GeometricProbs method)
(Gumbel method)
(HMC method)
(HMCECS method)
(HMCGibbs method)
(HalfCauchy method)
(HalfNormal method)
(Independent method)
(LKJCholesky method)
(Laplace method)
(LeftTruncatedDistribution method)
(Logistic method)
(LowRankMultivariateNormal method)
(MCMCKernel method)
(MaskedDistribution method)
(MixedHMC method)
(MultinomialLogits method)
(MultinomialProbs method)
(MultivariateNormal method)
(Normal method)
(PRNGIdentity method)
(Poisson method)
(ProjectedNormal method)
(RightTruncatedDistribution method)
(SA method)
(SoftLaplace method)
(StudentT method)
(TransformedDistribution method)
(TruncatedPolyaGamma method)
(TwoSidedTruncatedDistribution method)
(Uniform method)
(Unit method)
(VonMises method)
(Weibull method)
(in module numpyro.primitives)
sample_field (BarkerMH attribute)
(HMC attribute)
(HMCGibbs attribute)
(MCMCKernel attribute)
(SA attribute)
sample_kernel() (in module numpyro.infer.hmc.hmc)
sample_posterior() (AutoContinuous method)
(AutoDelta method)
(AutoGuide method)
(AutoLaplaceApproximation method)
(AutoNormal method)
sample_with_intermediates() (Distribution method)
(ExpandedDistribution method)
(TransformedDistribution method)
SAState (in module numpyro.infer.sa)
scale (class in numpyro.handlers)
scale_constraint (AutoDiagonalNormal attribute)
(AutoLowRankMultivariateNormal attribute)
(AutoNormal attribute)
scale_tril (LowRankMultivariateNormal attribute)
scale_tril_constraint (AutoMultivariateNormal attribute)
scan() (in module numpyro.contrib.control_flow)
scope (class in numpyro.handlers)
seed (class in numpyro.handlers)
set_default_validate_args() (Distribution static method)
set_host_device_count() (in module numpyro.util)
set_platform() (in module numpyro.util)
SGD (class in numpyro.optim)
shape() (Distribution method)
SigmoidBeta (class in numpyro.contrib.tfp.distributions)
SigmoidTransform (class in numpyro.distributions.transforms)
simplex (in module numpyro.distributions.constraints)
SinhArcsinh (class in numpyro.contrib.tfp.distributions)
Skellam (class in numpyro.contrib.tfp.distributions)
SliceSampler (class in numpyro.contrib.tfp.mcmc)
SM3 (class in numpyro.optim)
SoftLaplace (class in numpyro.distributions.continuous)
softplus_lower_cholesky (in module numpyro.distributions.constraints)
softplus_positive (in module numpyro.distributions.constraints)
SoftplusLowerCholeskyTransform (class in numpyro.distributions.transforms)
SoftplusTransform (class in numpyro.distributions.transforms)
sphere (in module numpyro.distributions.constraints)
SphericalUniform (class in numpyro.contrib.tfp.distributions)
split_gelman_rubin() (in module numpyro.diagnostics)
stable_update() (SVI method)
StickBreakingTransform (class in numpyro.distributions.transforms)
StoppingRatioLogistic (class in numpyro.contrib.tfp.distributions)
StudentT (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
StudentTProcess (class in numpyro.contrib.tfp.distributions)
subsample() (in module numpyro.primitives)
substitute (class in numpyro.handlers)
summary() (in module numpyro.diagnostics)
support (BernoulliLogits attribute)
(BernoulliProbs attribute)
(Beta attribute)
(BetaBinomial attribute)
(BetaProportion attribute)
(BinomialLogits attribute)
(BinomialProbs attribute)
(CategoricalLogits attribute)
(CategoricalProbs attribute)
(Cauchy attribute)
(Delta attribute)
(Dirichlet attribute)
(DirichletMultinomial attribute)
(Distribution attribute)
(ExpandedDistribution attribute)
(Exponential attribute)
(FoldedDistribution attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(GeometricLogits attribute)
(GeometricProbs attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(ImproperUniform attribute)
(Independent attribute)
(InverseGamma attribute)
(LKJ attribute)
(LKJCholesky attribute)
(Laplace attribute)
(LeftTruncatedDistribution attribute)
(LogNormal attribute)
(Logistic attribute)
(LowRankMultivariateNormal attribute)
(MaskedDistribution attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(NegativeBinomial2 attribute)
(NegativeBinomialLogits attribute)
(NegativeBinomialProbs attribute)
(Normal attribute)
(Pareto attribute)
(Poisson attribute)
(ProjectedNormal attribute)
(RightTruncatedDistribution attribute)
(SoftLaplace attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedPolyaGamma attribute)
(TwoSidedTruncatedDistribution attribute)
(Uniform attribute)
(Unit attribute)
(VonMises attribute)
(Weibull attribute)
(ZeroInflatedPoisson attribute)
supported_types (LeftTruncatedDistribution attribute)
(RightTruncatedDistribution attribute)
(TwoSidedTruncatedDistribution attribute)
SVI (class in numpyro.infer.svi)
T
taylor_proxy() (HMCECS static method)
(in module numpyro.infer.hmc_gibbs)
TFPDistribution (class in numpyro.contrib.tfp.distributions)
TFPKernel (class in numpyro.contrib.tfp.mcmc)
to_data() (in module numpyro.contrib.funsor.enum_messenger)
to_event() (Distribution method)
to_funsor() (in module numpyro.contrib.funsor.enum_messenger)
trace (class in numpyro.contrib.funsor.enum_messenger)
(class in numpyro.handlers)
Trace_ELBO (class in numpyro.infer.elbo)
TraceGraph_ELBO (class in numpyro.infer.elbo)
TraceMeanField_ELBO (class in numpyro.infer.elbo)
Transform (class in numpyro.distributions.transforms)
transform_fn() (in module numpyro.infer.util)
transform_sample() (NeuTraReparam method)
TransformedDistribution (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.distribution)
TransformReparam (class in numpyro.infer.reparam)
tree_flatten() (Delta method)
(Distribution method)
(ExpandedDistribution method)
(FoldedDistribution method)
(GaussianRandomWalk method)
(ImproperUniform method)
(Independent method)
(InverseGamma method)
(LKJ method)
(LKJCholesky method)
(LeftTruncatedDistribution method)
(LogNormal method)
(MaskedDistribution method)
(MultivariateNormal method)
(Pareto method)
(RightTruncatedDistribution method)
(TransformedDistribution method)
(TruncatedCauchy method)
(TruncatedNormal method)
(TruncatedPolyaGamma method)
(TwoSidedTruncatedDistribution method)
(Uniform method)
tree_unflatten() (numpyro.distributions.continuous.GaussianRandomWalk class method)
(numpyro.distributions.continuous.LKJ class method)
(numpyro.distributions.continuous.LKJCholesky class method)
(numpyro.distributions.continuous.MultivariateNormal class method)
(numpyro.distributions.continuous.Uniform class method)
(numpyro.distributions.distribution.Delta class method)
(numpyro.distributions.distribution.Distribution class method)
(numpyro.distributions.distribution.ExpandedDistribution class method)
(numpyro.distributions.distribution.FoldedDistribution class method)
(numpyro.distributions.distribution.Independent class method)
(numpyro.distributions.distribution.MaskedDistribution class method)
(numpyro.distributions.truncated.LeftTruncatedDistribution class method)
(numpyro.distributions.truncated.RightTruncatedDistribution class method)
(numpyro.distributions.truncated.TruncatedCauchy class method)
(numpyro.distributions.truncated.TruncatedNormal class method)
(numpyro.distributions.truncated.TruncatedPolyaGamma class method)
(numpyro.distributions.truncated.TwoSidedTruncatedDistribution class method)
Triangular (class in numpyro.contrib.tfp.distributions)
TruncatedCauchy (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.truncated)
TruncatedDistribution() (in module numpyro.distributions.truncated)
TruncatedNormal (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.truncated)
TruncatedPolyaGamma (class in numpyro.distributions.truncated)
truncation_point (TruncatedPolyaGamma attribute)
TwoSidedTruncatedDistribution (class in numpyro.distributions.truncated)
U
UncalibratedHamiltonianMonteCarlo (class in numpyro.contrib.tfp.mcmc)
UncalibratedLangevin (class in numpyro.contrib.tfp.mcmc)
UncalibratedRandomWalk (class in numpyro.contrib.tfp.mcmc)
Uniform (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
Unit (class in numpyro.distributions.distribution)
unit_interval (in module numpyro.distributions.constraints)
update() (Adagrad method)
(Adam method)
(ClippedAdam method)
(Minimize method)
(Momentum method)
(RMSProp method)
(RMSPropMomentum method)
(SGD method)
(SM3 method)
(SVI method)
V
validation_enabled() (in module numpyro.distributions.distribution)
variance (BernoulliLogits attribute)
(BernoulliProbs attribute)
(Beta attribute)
(BetaBinomial attribute)
(BinomialLogits attribute)
(BinomialProbs attribute)
(CategoricalLogits attribute)
(CategoricalProbs attribute)
(Cauchy attribute)
(Delta attribute)
(Dirichlet attribute)
(DirichletMultinomial attribute)
(Distribution attribute)
(ExpandedDistribution attribute)
(Exponential attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(GeometricLogits attribute)
(GeometricProbs attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(Independent attribute)
(InverseGamma attribute)
(Laplace attribute)
(LogNormal attribute)
(Logistic attribute)
(LowRankMultivariateNormal attribute)
(MaskedDistribution attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(Poisson attribute)
(SoftLaplace attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
(VonMises attribute)
(Weibull attribute)
VariationalGaussianProcess (class in numpyro.contrib.tfp.distributions)
VectorDeterministic (class in numpyro.contrib.tfp.distributions)
VectorExponentialDiag (class in numpyro.contrib.tfp.distributions)
Vindex (class in numpyro.contrib.indexing)
vindex() (in module numpyro.contrib.indexing)
VonMises (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.directional)
VonMisesFisher (class in numpyro.contrib.tfp.distributions)
W
warmup() (MCMC method)
Weibull (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
WishartLinearOperator (class in numpyro.contrib.tfp.distributions)
WishartTriL (class in numpyro.contrib.tfp.distributions)
Z
ZeroInflatedDistribution() (in module numpyro.distributions.discrete)
ZeroInflatedNegativeBinomial2() (in module numpyro.distributions.conjugate)
ZeroInflatedPoisson (class in numpyro.distributions.discrete)
Zipf (class in numpyro.contrib.tfp.distributions)
Read the Docs
v: 0.7.0
Versions
latest
stable
0.7.0
0.6.0
0.5.0
0.4.1
0.4.0
0.3.0
v0.2.4
v0.2.3
v0.2.2
v0.2.1
v0.2.0
v0.1.0
fix-docs
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.