NumPyro
0.6.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: Hamiltonian Monte Carlo with Energy Conserving Subsampling
Example: Hidden Markov Model
Example: Predator-Prey Model
Example: Neural Transport
Example: MCMC Methods for Tall Data
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)
(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)
(Normal attribute)
(OrderedLogistic attribute)
(Pareto attribute)
(Poisson attribute)
(ProjectedNormal attribute)
(RightTruncatedDistribution attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(TruncatedPolyaGamma attribute)
(TwoSidedTruncatedDistribution 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
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)
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)
(Laplace method)
(Logistic method)
(Normal method)
(StudentT 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)
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)
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)
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_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() (Cauchy method)
(Distribution method)
(Laplace method)
(Logistic method)
(Normal method)
(StudentT 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_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 (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)
LeftTruncatedDistribution (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)
(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)
(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)
(Normal method)
(Poisson method)
(ProjectedNormal method)
(RightTruncatedDistribution method)
(StudentT method)
(TransformedDistribution method)
(TruncatedPolyaGamma method)
(TwoSidedTruncatedDistribution method)
(Uniform 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)
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)
(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)
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)
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)
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)
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)
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() (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)
(LeftTruncatedDistribution attribute)
(LogNormal attribute)
(Logistic attribute)
(LowRankMultivariateNormal attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(ProjectedNormal attribute)
(RightTruncatedDistribution attribute)
(StudentT attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(TwoSidedTruncatedDistribution attribute)
(Uniform attribute)
(VonMises attribute)
replay (class in numpyro.handlers)
ReplicaExchangeMC (class in numpyro.contrib.tfp.mcmc)
RightTruncatedDistribution (class in numpyro.distributions.continuous)
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() (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)
(StudentT method)
(TransformedDistribution method)
(TruncatedPolyaGamma method)
(TwoSidedTruncatedDistribution method)
(Uniform method)
(Unit method)
(VonMises method)
(ZeroInflatedPoisson 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)
(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)
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)
(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)
(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)
(Normal attribute)
(Pareto attribute)
(Poisson attribute)
(ProjectedNormal attribute)
(RightTruncatedDistribution attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedPolyaGamma attribute)
(TwoSidedTruncatedDistribution attribute)
(Uniform attribute)
(Unit attribute)
(VonMises 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)
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)
(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.LeftTruncatedDistribution class method)
(numpyro.distributions.continuous.MultivariateNormal class method)
(numpyro.distributions.continuous.RightTruncatedDistribution class method)
(numpyro.distributions.continuous.TruncatedCauchy class method)
(numpyro.distributions.continuous.TruncatedNormal class method)
(numpyro.distributions.continuous.TruncatedPolyaGamma class method)
(numpyro.distributions.continuous.TwoSidedTruncatedDistribution 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)
TruncatedDistribution() (in module 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)
TwoSidedTruncatedDistribution (class in numpyro.distributions.continuous)
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)
Zipf (class in numpyro.contrib.tfp.distributions)
Read the Docs
v: 0.6.0
Versions
latest
stable
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
.