NumPyro
v0.2.2
Modeling:
Pyro Primitives
Effect Handlers
Distributions:
Base Distribution
Continuous Distributions
Discrete Distributions
Constraints
Transforms
Flows
Inference:
Markov Chain Monte Carlo (MCMC)
Stochastic Variational Inference (SVI)
Automatic Guide Generation
Optimizers
Diagnostics
Runtime Utilities
Inference Utilities
NumPyro
Docs
»
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
Z
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)
(BinomialLogits attribute)
(BinomialProbs attribute)
(CategoricalLogits attribute)
(CategoricalProbs attribute)
(Cauchy attribute)
(Chi2 attribute)
(Delta attribute)
(Dirichlet attribute)
(Distribution attribute)
(Exponential attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(Independent attribute)
(InverseGamma attribute)
(LKJ attribute)
(LKJCholesky attribute)
(LogNormal attribute)
(LowRankMultivariateNormal attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(Normal attribute)
(OrderedLogistic attribute)
(Pareto attribute)
(Poisson attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
(Unit attribute)
(ZeroInflatedPoisson attribute)
AutoContinuous (class in numpyro.contrib.autoguide)
AutoContinuousELBO (class in numpyro.contrib.autoguide)
autocorrelation() (in module numpyro.diagnostics)
autocovariance() (in module numpyro.diagnostics)
AutoDiagonalNormal (class in numpyro.contrib.autoguide)
AutoIAFNormal (class in numpyro.contrib.autoguide)
AutoLaplaceApproximation (class in numpyro.contrib.autoguide)
AutoMultivariateNormal (class in numpyro.contrib.autoguide)
B
base_dist (AutoContinuous attribute)
batch_shape (Distribution attribute)
Bernoulli() (in module numpyro.distributions.discrete)
BernoulliLogits (class in numpyro.distributions.discrete)
BernoulliProbs (class in numpyro.distributions.discrete)
Beta (class in numpyro.distributions.continuous)
BetaBinomial (class in numpyro.distributions.conjugate)
biject_to() (in module numpyro.distributions.transforms)
Binomial() (in module numpyro.distributions.discrete)
BinomialLogits (class in numpyro.distributions.discrete)
BinomialProbs (class in numpyro.distributions.discrete)
block (class in numpyro.handlers)
boolean (in module numpyro.distributions.constraints)
C
call_with_intermediates() (ComposeTransform method)
(InverseAutoregressiveTransform method)
(Transform method)
Categorical() (in module numpyro.distributions.discrete)
CategoricalLogits (class in numpyro.distributions.discrete)
CategoricalProbs (class in numpyro.distributions.discrete)
Cauchy (class in numpyro.distributions.continuous)
Chi2 (class in numpyro.distributions.continuous)
ClippedAdam (class in numpyro.optim)
codomain (AbsTransform attribute)
(AffineTransform attribute)
(ComposeTransform attribute)
(CorrCholeskyTransform attribute)
(ExpTransform attribute)
(InvCholeskyTransform attribute)
(InverseAutoregressiveTransform attribute)
(LowerCholeskyTransform attribute)
(MultivariateAffineTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(PowerTransform attribute)
(SigmoidTransform attribute)
(StickBreakingTransform attribute)
(Transform attribute)
ComposeTransform (class in numpyro.distributions.transforms)
condition (class in numpyro.handlers)
consensus() (in module numpyro.infer.hmc_util)
constrain_fn() (HMC method)
(in module numpyro.infer.util)
corr_cholesky (in module numpyro.distributions.constraints)
corr_matrix (in module numpyro.distributions.constraints)
CorrCholeskyTransform (class in numpyro.distributions.transforms)
covariance_matrix (LowRankMultivariateNormal attribute)
(MultivariateNormal attribute)
D
Delta (class in numpyro.distributions.discrete)
dependent (in module numpyro.distributions.constraints)
Dirichlet (class in numpyro.distributions.continuous)
Distribution (class in numpyro.distributions.distribution)
domain (AbsTransform attribute)
(ComposeTransform attribute)
(CorrCholeskyTransform attribute)
(InverseAutoregressiveTransform attribute)
(LowerCholeskyTransform attribute)
(MultivariateAffineTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(PowerTransform attribute)
(StickBreakingTransform attribute)
(Transform attribute)
E
effective_sample_size() (in module numpyro.diagnostics)
ELBO (class in numpyro.infer.elbo)
enable_validation() (in module numpyro.distributions.distribution)
enable_x64() (in module numpyro.util)
entropy() (LowRankMultivariateNormal method)
evaluate() (SVI method)
event_dim (AffineTransform attribute)
(ComposeTransform attribute)
(CorrCholeskyTransform attribute)
(InvCholeskyTransform attribute)
(InverseAutoregressiveTransform attribute)
(LowerCholeskyTransform attribute)
(MultivariateAffineTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(StickBreakingTransform attribute)
(Transform attribute)
event_shape (Distribution attribute)
Exponential (class in numpyro.distributions.continuous)
ExpTransform (class in numpyro.distributions.transforms)
F
factor() (in module numpyro.primitives)
find_valid_initial_params() (in module numpyro.infer.util)
fori_collect() (in module numpyro.util)
G
Gamma (class in numpyro.distributions.continuous)
GammaPoisson (class in numpyro.distributions.conjugate)
GaussianRandomWalk (class in numpyro.distributions.continuous)
gelman_rubin() (in module numpyro.diagnostics)
get_extra_fields() (MCMC method)
get_params() (Adagrad method)
(Adam method)
(ClippedAdam method)
(Momentum method)
(RMSProp method)
(RMSPropMomentum method)
(SGD method)
(SM3 method)
(SVI method)
get_samples() (MCMC method)
(Predictive method)
get_trace() (trace method)
get_transform() (AutoContinuous method)
(AutoLaplaceApproximation method)
greater_than() (in module numpyro.distributions.constraints)
H
HalfCauchy (class in numpyro.distributions.continuous)
HalfNormal (class in numpyro.distributions.continuous)
HMC (class in numpyro.infer.mcmc)
hmc() (in module numpyro.infer.mcmc)
HMCState (in module numpyro.infer.mcmc)
hpdi() (in module numpyro.diagnostics)
I
icdf() (Normal method)
IdentityTransform (class in numpyro.distributions.transforms)
Independent (class in numpyro.distributions.distribution)
init() (Adagrad method)
(Adam method)
(ClippedAdam method)
(HMC method)
(Momentum method)
(RMSProp method)
(RMSPropMomentum method)
(SGD method)
(SM3 method)
(SVI method)
init_kernel() (in module numpyro.infer.mcmc.hmc)
init_to_feasible() (in module numpyro.infer.util)
init_to_median() (in module numpyro.infer.util)
init_to_prior() (in module numpyro.infer.util)
init_to_uniform() (in module numpyro.infer.util)
init_to_value() (in module numpyro.infer.util)
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() (AbsTransform method)
(AffineTransform method)
(ComposeTransform method)
(CorrCholeskyTransform method)
(ExpTransform method)
(IdentityTransform method)
(InvCholeskyTransform method)
(InverseAutoregressiveTransform method)
(LowerCholeskyTransform method)
(MultivariateAffineTransform method)
(OrderedTransform method)
(PermuteTransform method)
(PowerTransform method)
(SigmoidTransform method)
(StickBreakingTransform method)
(Transform method)
InvCholeskyTransform (class in numpyro.distributions.transforms)
InverseAutoregressiveTransform (class in numpyro.distributions.flows)
InverseGamma (class in numpyro.distributions.continuous)
L
LKJ (class in numpyro.distributions.continuous)
LKJCholesky (class in numpyro.distributions.continuous)
log_abs_det_jacobian() (AffineTransform method)
(ComposeTransform method)
(CorrCholeskyTransform method)
(ExpTransform method)
(IdentityTransform method)
(InvCholeskyTransform method)
(InverseAutoregressiveTransform method)
(LowerCholeskyTransform method)
(MultivariateAffineTransform method)
(OrderedTransform method)
(PermuteTransform method)
(PowerTransform method)
(SigmoidTransform method)
(StickBreakingTransform method)
(Transform method)
log_density() (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)
(Distribution method)
(Exponential method)
(Gamma method)
(GammaPoisson method)
(GaussianRandomWalk method)
(HalfCauchy method)
(HalfNormal method)
(Independent method)
(LKJCholesky method)
(LowRankMultivariateNormal method)
(MultinomialLogits method)
(MultinomialProbs method)
(MultivariateNormal method)
(Normal method)
(Poisson method)
(StudentT method)
(TransformedDistribution method)
(Unit method)
(ZeroInflatedPoisson method)
LogNormal (class in numpyro.distributions.continuous)
loss() (AutoContinuousELBO method)
(ELBO method)
(RenyiELBO method)
lower_cholesky (in module numpyro.distributions.constraints)
LowerCholeskyTransform (class in numpyro.distributions.transforms)
LowRankMultivariateNormal (class in numpyro.distributions.continuous)
M
MCMC (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)
(Distribution attribute)
(Exponential attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(Independent attribute)
(InverseGamma attribute)
(LKJ attribute)
(LogNormal attribute)
(LowRankMultivariateNormal attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(Poisson attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
(ZeroInflatedPoisson attribute)
median() (AutoContinuous method)
(AutoDiagonalNormal method)
(AutoLaplaceApproximation method)
(AutoMultivariateNormal method)
model (HMC attribute)
module() (in module numpyro.primitives)
Momentum (class in numpyro.optim)
multinomial() (in module numpyro.distributions.constraints)
Multinomial() (in module numpyro.distributions.discrete)
MultinomialLogits (class in numpyro.distributions.discrete)
MultinomialProbs (class in numpyro.distributions.discrete)
MultivariateAffineTransform (class in numpyro.distributions.transforms)
MultivariateNormal (class in numpyro.distributions.continuous)
N
nonnegative_integer (in module numpyro.distributions.constraints)
Normal (class in numpyro.distributions.continuous)
numpyro.contrib.autoguide (module)
numpyro.diagnostics (module)
numpyro.handlers (module)
numpyro.infer.util (module)
numpyro.optim (module)
numpyro.primitives (module)
numpyro.util (module)
NUTS (class in numpyro.infer.mcmc)
O
ordered_vector (in module numpyro.distributions.constraints)
OrderedLogistic (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.distributions.continuous)
PermuteTransform (class in numpyro.distributions.transforms)
plate (class in numpyro.primitives)
Poisson (class in numpyro.distributions.discrete)
positive (in module numpyro.distributions.constraints)
positive_definite (in module numpyro.distributions.constraints)
positive_integer (in module numpyro.distributions.constraints)
postprocess_message() (trace method)
potential_energy() (in module numpyro.infer.util)
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)
PRNGIdentity (class in numpyro.distributions.discrete)
probs (BernoulliLogits attribute)
(BinomialLogits attribute)
(CategoricalLogits attribute)
(MultinomialLogits attribute)
process_message() (block method)
(condition method)
(replay method)
(scale method)
(seed method)
(substitute method)
Q
quantiles() (AutoContinuous method)
(AutoDiagonalNormal method)
(AutoLaplaceApproximation method)
(AutoMultivariateNormal method)
R
real (in module numpyro.distributions.constraints)
real_vector (in module numpyro.distributions.constraints)
RenyiELBO (class in numpyro.infer.elbo)
reparameterized_params (Independent attribute)
reparametrized_params (Cauchy attribute)
(Distribution attribute)
(Exponential attribute)
(Gamma attribute)
(GaussianRandomWalk attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(InverseGamma attribute)
(LogNormal attribute)
(MultivariateNormal attribute)
(Normal attribute)
(StudentT attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
replay (class in numpyro.handlers)
RMSProp (class in numpyro.optim)
RMSPropMomentum (class in numpyro.optim)
run() (MCMC method)
S
sample() (BernoulliLogits method)
(BernoulliProbs method)
(Beta method)
(BetaBinomial method)
(BinomialLogits method)
(BinomialProbs method)
(CategoricalLogits method)
(CategoricalProbs method)
(Cauchy method)
(Delta method)
(Dirichlet method)
(Distribution method)
(Exponential method)
(Gamma method)
(GammaPoisson method)
(GaussianRandomWalk method)
(HMC method)
(HalfCauchy method)
(HalfNormal method)
(Independent method)
(LKJCholesky method)
(LowRankMultivariateNormal method)
(MultinomialLogits method)
(MultinomialProbs method)
(MultivariateNormal method)
(Normal method)
(PRNGIdentity method)
(Poisson method)
(StudentT method)
(TransformedDistribution method)
(Unit method)
(ZeroInflatedPoisson method)
(in module numpyro.primitives)
sample_kernel() (in module numpyro.infer.mcmc.hmc)
sample_posterior() (AutoContinuous method)
(AutoLaplaceApproximation method)
sample_with_intermediates() (Distribution method)
(TransformedDistribution method)
scale (class in numpyro.handlers)
scale_tril (LowRankMultivariateNormal attribute)
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)
SigmoidTransform (class in numpyro.distributions.transforms)
simplex (in module numpyro.distributions.constraints)
SM3 (class in numpyro.optim)
split_gelman_rubin() (in module numpyro.diagnostics)
StickBreakingTransform (class in numpyro.distributions.transforms)
StudentT (class in numpyro.distributions.continuous)
substitute (class in numpyro.handlers)
summary() (in module numpyro.diagnostics)
support (BernoulliLogits attribute)
(BernoulliProbs attribute)
(Beta attribute)
(BetaBinomial attribute)
(BinomialLogits attribute)
(BinomialProbs attribute)
(CategoricalLogits attribute)
(CategoricalProbs attribute)
(Cauchy attribute)
(Delta attribute)
(Dirichlet attribute)
(Distribution attribute)
(Exponential attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(Independent attribute)
(InverseGamma attribute)
(LKJ attribute)
(LKJCholesky attribute)
(LowRankMultivariateNormal attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(Poisson attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(Unit attribute)
(ZeroInflatedPoisson attribute)
SVI (class in numpyro.infer.svi)
T
to_event() (Distribution method)
trace (class in numpyro.handlers)
Transform (class in numpyro.distributions.transforms)
transform_fn() (in module numpyro.infer.util)
transform_with_intermediates() (Distribution method)
(TransformedDistribution method)
TransformedDistribution (class in numpyro.distributions.distribution)
TruncatedCauchy (class in numpyro.distributions.continuous)
TruncatedNormal (class in numpyro.distributions.continuous)
U
Uniform (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)
(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)
(Distribution attribute)
(Exponential attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(Independent attribute)
(InverseGamma attribute)
(LKJ attribute)
(LogNormal attribute)
(LowRankMultivariateNormal attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(Poisson attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
(ZeroInflatedPoisson attribute)
W
warmup() (MCMC method)
Z
ZeroInflatedPoisson (class in numpyro.distributions.discrete)
Read the Docs
v: v0.2.2
Versions
latest
stable
v0.2.2
v0.2.1
v0.2.0
v0.1.0
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.