NumPyro
0.4.1
Modeling:
Pyro Primitives
Effect Handlers
Distributions:
Base Distribution
Continuous Distributions
Discrete Distributions
Directional Distributions
TensorFlow Distributions
Constraints
Transforms
Flows
Inference:
Markov Chain Monte Carlo (MCMC)
Stochastic Variational Inference (SVI)
Automatic Guide Generation
Reparameterizers
Funsor-based NumPyro
Optimizers
Diagnostics
Runtime Utilities
Inference Utilities
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)
(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)
(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)
(LogNormal attribute)
(Logistic attribute)
(LowRankMultivariateNormal attribute)
(MaskedDistribution attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(Normal attribute)
(OrderedLogistic attribute)
(Pareto attribute)
(Poisson attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(TruncatedPolyaGamma attribute)
(Uniform attribute)
(Unit attribute)
(VonMises 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)
AutoDiagonalNormal (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
batch_shape (Distribution attribute)
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)
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)
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)
ClippedAdam (class in numpyro.optim)
codomain (AbsTransform attribute)
(AffineTransform attribute)
(ComposeTransform attribute)
(CorrCholeskyTransform attribute)
(ExpTransform attribute)
(InvCholeskyTransform attribute)
(InverseAutoregressiveTransform attribute)
(LowerCholeskyAffine attribute)
(LowerCholeskyTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(PowerTransform attribute)
(SigmoidTransform attribute)
(StickBreakingTransform attribute)
(Transform attribute)
collapse (class in numpyro.handlers)
ComposeTransform (class in numpyro.distributions.transforms)
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)
covariance_matrix (LowRankMultivariateNormal attribute)
(MultivariateNormal attribute)
D
default_fields (HMC attribute)
(MCMCKernel attribute)
(SA attribute)
Delta (class in numpyro.distributions.discrete)
dependent (in module numpyro.distributions.constraints)
Deterministic (class in numpyro.contrib.tfp.distributions)
deterministic() (in module numpyro.primitives)
Dirichlet (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
DirichletMultinomial (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.conjugate)
Distribution (class in numpyro.distributions.distribution)
do (class in numpyro.handlers)
domain (AbsTransform attribute)
(ComposeTransform attribute)
(CorrCholeskyTransform attribute)
(InverseAutoregressiveTransform attribute)
(LowerCholeskyAffine attribute)
(LowerCholeskyTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(PowerTransform 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_update() (Adagrad method)
(Adam method)
(ClippedAdam method)
(Minimize method)
(Momentum method)
(RMSProp method)
(RMSPropMomentum method)
(SGD method)
(SM3 method)
evaluate() (SVI method)
event_dim (AffineTransform attribute)
(BlockNeuralAutoregressiveTransform attribute)
(ComposeTransform attribute)
(CorrCholeskyTransform attribute)
(Distribution attribute)
(InvCholeskyTransform attribute)
(InverseAutoregressiveTransform attribute)
(LowerCholeskyAffine attribute)
(LowerCholeskyTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(StickBreakingTransform 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)
ExpRelaxedOneHotCategorical (class in numpyro.contrib.tfp.distributions)
ExpTransform (class in numpyro.distributions.transforms)
F
factor() (in module numpyro.primitives)
find_valid_initial_params() (in module numpyro.infer.util)
FiniteDiscrete (class in numpyro.contrib.tfp.distributions)
flax_module() (in module numpyro.contrib.module)
fori_collect() (in module numpyro.util)
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)
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() (HMC method)
(MCMCKernel method)
(SA method)
get_extra_fields() (MCMC method)
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)
get_trace() (trace method)
get_transform() (AutoContinuous method)
(AutoDiagonalNormal method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal 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)
HiddenMarkovModel (class in numpyro.contrib.tfp.distributions)
HMC (class in numpyro.infer.hmc)
hmc() (in module numpyro.infer.hmc)
HMCState (in module numpyro.infer.hmc)
Horseshoe (class in numpyro.contrib.tfp.distributions)
hpdi() (in module numpyro.diagnostics)
I
icdf() (Normal 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)
init() (Adagrad method)
(Adam method)
(ClippedAdam method)
(HMC method)
(MCMCKernel method)
(Minimize 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() (AbsTransform method)
(AffineTransform method)
(BlockNeuralAutoregressiveTransform method)
(ComposeTransform method)
(CorrCholeskyTransform method)
(ExpTransform method)
(IdentityTransform method)
(InvCholeskyTransform method)
(InverseAutoregressiveTransform method)
(LowerCholeskyAffine method)
(LowerCholeskyTransform 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.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
InverseGaussian (class in numpyro.contrib.tfp.distributions)
is_discrete (BernoulliLogits attribute)
(BernoulliProbs attribute)
(BetaBinomial attribute)
(BinomialLogits attribute)
(BinomialProbs attribute)
(CategoricalLogits attribute)
(CategoricalProbs attribute)
(Delta attribute)
(DirichletMultinomial attribute)
(Distribution attribute)
(ExpandedDistribution attribute)
(GammaPoisson attribute)
(GeometricLogits attribute)
(GeometricProbs attribute)
(Independent attribute)
(MaskedDistribution attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(PRNGIdentity attribute)
(Poisson attribute)
(ZeroInflatedPoisson 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
Laplace (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
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)
(ComposeTransform method)
(CorrCholeskyTransform method)
(ExpTransform method)
(IdentityTransform method)
(InvCholeskyTransform method)
(InverseAutoregressiveTransform method)
(LowerCholeskyAffine method)
(LowerCholeskyTransform method)
(OrderedTransform method)
(PermuteTransform method)
(PowerTransform method)
(SigmoidTransform 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)
(Gamma method)
(GammaPoisson method)
(GaussianRandomWalk method)
(GeometricLogits method)
(GeometricProbs method)
(Gumbel method)
(HalfCauchy method)
(HalfNormal method)
(ImproperUniform method)
(Independent method)
(LKJCholesky method)
(Laplace method)
(Logistic method)
(LowRankMultivariateNormal method)
(MaskedDistribution method)
(MultinomialLogits method)
(MultinomialProbs method)
(MultivariateNormal method)
(Normal method)
(Poisson method)
(StudentT method)
(TransformedDistribution method)
(TruncatedPolyaGamma method)
(Unit method)
(VonMises method)
(ZeroInflatedPoisson method)
Logistic (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
LogitNormal (class in numpyro.contrib.tfp.distributions)
LogLogistic (class in numpyro.contrib.tfp.distributions)
LogNormal (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
loss() (RenyiELBO 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 (class in numpyro.contrib.funsor.enum_messenger)
mask (class in numpyro.handlers)
mask() (Distribution method)
MaskedDistribution (class in numpyro.distributions.distribution)
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)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
(VonMises attribute)
(ZeroInflatedPoisson attribute)
median() (AutoContinuous method)
(AutoDiagonalNormal method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
(AutoNormal method)
MetropolisAdjustedLangevinAlgorithm (class in numpyro.contrib.tfp.mcmc)
Minimize (class in numpyro.optim)
MixtureSameFamily (class in numpyro.contrib.tfp.distributions)
model (HMC 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)
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)
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)
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)
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)
postprocess_fn() (HMC 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)
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)
(lift method)
(mask method)
(plate method)
(reparam method)
(replay method)
(scale method)
(scope method)
(seed method)
(substitute method)
Q
quantiles() (AutoContinuous method)
(AutoDiagonalNormal 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)
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 (Cauchy attribute)
(Distribution attribute)
(Exponential attribute)
(Gamma attribute)
(GaussianRandomWalk attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(InverseGamma attribute)
(Laplace attribute)
(LogNormal attribute)
(Logistic attribute)
(MultivariateNormal attribute)
(Normal attribute)
(StudentT attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
replay (class in numpyro.handlers)
ReplicaExchangeMC (class in numpyro.contrib.tfp.mcmc)
RMSProp (class in numpyro.optim)
RMSPropMomentum (class in numpyro.optim)
run() (MCMC method)
S
SA (class in numpyro.infer.sa)
Sample (class in numpyro.contrib.tfp.distributions)
sample() (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)
(Gamma method)
(GammaPoisson method)
(GaussianRandomWalk method)
(GeometricLogits method)
(GeometricProbs method)
(Gumbel method)
(HMC method)
(HalfCauchy method)
(HalfNormal method)
(Independent method)
(LKJCholesky method)
(Laplace method)
(Logistic method)
(LowRankMultivariateNormal method)
(MCMCKernel method)
(MaskedDistribution method)
(MultinomialLogits method)
(MultinomialProbs method)
(MultivariateNormal method)
(Normal method)
(PRNGIdentity method)
(Poisson method)
(SA method)
(StudentT method)
(TransformedDistribution method)
(TruncatedPolyaGamma method)
(Unit method)
(VonMises method)
(ZeroInflatedPoisson method)
(in module numpyro.primitives)
sample_field (HMC attribute)
(MCMCKernel attribute)
(SA attribute)
sample_kernel() (in module numpyro.infer.hmc.hmc)
sample_posterior() (AutoContinuous method)
(AutoLaplaceApproximation method)
(AutoNormal method)
sample_with_intermediates() (Distribution method)
(TransformedDistribution method)
SAState (in module numpyro.infer.sa)
scale (class in numpyro.handlers)
scale_tril (LowRankMultivariateNormal 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)
SigmoidTransform (class in numpyro.distributions.transforms)
simplex (in module numpyro.distributions.constraints)
SinhArcsinh (class in numpyro.contrib.tfp.distributions)
SliceSampler (class in numpyro.contrib.tfp.mcmc)
SM3 (class in numpyro.optim)
SphericalUniform (class in numpyro.contrib.tfp.distributions)
split_gelman_rubin() (in module numpyro.diagnostics)
StickBreakingTransform (class in numpyro.distributions.transforms)
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)
(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)
(LKJCholesky attribute)
(Laplace attribute)
(Logistic attribute)
(LowRankMultivariateNormal attribute)
(MaskedDistribution attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(Poisson attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(TruncatedPolyaGamma attribute)
(Uniform attribute)
(Unit attribute)
(VonMises attribute)
(ZeroInflatedPoisson attribute)
SVI (class in numpyro.infer.svi)
T
TFPDistributionMixin (class in numpyro.contrib.tfp.distributions)
TFPKernel (class in numpyro.contrib.tfp.mcmc)
to_data (class in numpyro.contrib.funsor.enum_messenger)
to_event() (Distribution method)
to_funsor (class in numpyro.contrib.funsor.enum_messenger)
trace (class in numpyro.contrib.funsor.enum_messenger)
(class in numpyro.handlers)
Trace_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)
(GaussianRandomWalk method)
(ImproperUniform method)
(Independent method)
(InverseGamma method)
(LKJ method)
(LKJCholesky method)
(LogNormal method)
(MaskedDistribution method)
(MultivariateNormal method)
(Pareto method)
(TransformedDistribution method)
(TruncatedCauchy method)
(TruncatedNormal method)
(TruncatedPolyaGamma 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.TruncatedCauchy class method)
(numpyro.distributions.continuous.TruncatedNormal class method)
(numpyro.distributions.continuous.TruncatedPolyaGamma class method)
(numpyro.distributions.continuous.Uniform class method)
(numpyro.distributions.discrete.Delta class method)
(numpyro.distributions.distribution.Distribution class method)
(numpyro.distributions.distribution.ExpandedDistribution class method)
(numpyro.distributions.distribution.Independent class method)
(numpyro.distributions.distribution.MaskedDistribution class method)
Triangular (class in numpyro.contrib.tfp.distributions)
TruncatedCauchy (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
TruncatedNormal (class in numpyro.contrib.tfp.distributions)
(class in numpyro.distributions.continuous)
TruncatedPolyaGamma (class in numpyro.distributions.continuous)
truncation_point (TruncatedPolyaGamma attribute)
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)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
(VonMises attribute)
(ZeroInflatedPoisson 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)
WishartLinearOperator (class in numpyro.contrib.tfp.distributions)
WishartTriL (class in numpyro.contrib.tfp.distributions)
Z
ZeroInflatedPoisson (class in numpyro.distributions.discrete)
Read the Docs
v: 0.4.1
Versions
latest
stable
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
.