NumPyro
0.5.0
Getting Started with NumPyro
API and Developer Reference
API Reference
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
Discrete Latent Variables
Example: Bayesian Models of Annotation
Example: Enumerate Hidden Markov Model
Example: CJS Capture-Recapture Model for Ecological Data
Bayesian Imputation for Missing Values in Discrete Covariates
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: Hidden Markov Model
Example: Predator-Prey Model
Example: Neural Transport
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)
AutoDelta (class in numpyro.infer.autoguide)
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)
(BlockNeuralAutoregressiveTransform 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.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)
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)
(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 (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)
fori_collect() (in module numpyro.util)
forward_shape() (AffineTransform method)
(ComposeTransform method)
(CorrCholeskyTransform method)
(LowerCholeskyAffine method)
(LowerCholeskyTransform method)
(PowerTransform 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() (HMC method)
(HMCGibbs 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)
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() (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)
(class in numpyro.handlers)
init() (Adagrad method)
(Adam method)
(ClippedAdam method)
(DiscreteHMCGibbs method)
(HMC method)
(HMCECS method)
(HMCGibbs 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 (Transform attribute)
InvCholeskyTransform (class in numpyro.distributions.transforms)
inverse_shape() (AffineTransform method)
(ComposeTransform method)
(CorrCholeskyTransform method)
(LowerCholeskyAffine method)
(LowerCholeskyTransform method)
(PowerTransform 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 (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)
last_state (MCMC attribute)
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)
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() (RenyiELBO 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)
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)
(AutoDelta 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)
(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)
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)
post_warmup_state (MCMC attribute)
postprocess_fn() (HMC 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)
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)
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 (Delta attribute)
(Independent attribute)
reparametrized_params (Beta attribute)
(Cauchy attribute)
(Chi2 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)
(LogNormal attribute)
(Logistic attribute)
(LowRankMultivariateNormal attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(StudentT attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(Uniform attribute)
(VonMises attribute)
replay (class in numpyro.handlers)
ReplicaExchangeMC (class in numpyro.contrib.tfp.mcmc)
RMSProp (class in numpyro.optim)
RMSPropMomentum (class in numpyro.optim)
rsample() (Distribution method)
(ExpandedDistribution method)
(Independent method)
(MaskedDistribution method)
(TransformedDistribution method)
run() (MCMC method)
(SVI 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)
(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)
(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)
(HMCGibbs attribute)
(MCMCKernel attribute)
(SA attribute)
sample_kernel() (in module numpyro.infer.hmc.hmc)
sample_posterior() (AutoContinuous method)
(AutoDelta 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)
Skellam (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)
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)
(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() (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)
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)
(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.distribution.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.5.0
Versions
latest
stable
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
.