NumPyro
0.8.0
Getting Started with NumPyro
API and Developer Reference
Pyro Primitives
Distributions
Inference
Effect Handlers
Contributed Code
Change Log
Introductory Tutorials
Bayesian Regression Using NumPyro
Bayesian Hierarchical Linear Regression
Example: Baseball Batting Average
Example: Variational Autoencoder
Example: Neal’s Funnel
Example: Stochastic Volatility
Example: ProdLDA with Flax and Haiku
Automatic rendering of NumPyro models
Bad posterior geometry and how to deal with it
Discrete Latent Variables
Example: Bayesian Models of Annotation
Example: Enumerate Hidden Markov Model
Example: CJS Capture-Recapture Model for Ecological Data
Example: Nested Sampling for Gaussian Shells
Bayesian Imputation for Missing Values in Discrete Covariates
Applications
Time Series Forecasting
Ordinal Regression
Bayesian Imputation
Example: Gaussian Process
Example: Bayesian Neural Network
Example: AutoDAIS
Example: Sparse Regression
Example: Horseshoe Regression
Example: Proportion Test
Example: Generalized Linear Mixed Models
Example: Hamiltonian Monte Carlo with Energy Conserving Subsampling
Example: Hidden Markov Model
Example: Hilbert space approximation for Gaussian processes.
Example: Predator-Prey Model
Example: Neural Transport
Example: MCMC Methods for Tall Data
Example: Thompson sampling for Bayesian Optimization with GPs
NumPyro
»
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
Z
_
__call__() (CircularReparam method)
(LocScaleReparam method)
(NeuTraReparam method)
(ProjectedNormalReparam method)
(TransformReparam method)
A
AbsTransform (class in numpyro.distributions.transforms)
Adagrad (class in numpyro.optim)
Adam (class in numpyro.optim)
AffineTransform (class in numpyro.distributions.transforms)
arg_constraints (BernoulliLogits attribute)
(BernoulliProbs attribute)
(Beta attribute)
(BetaBinomial attribute)
(BetaProportion attribute)
(BinomialLogits attribute)
(BinomialProbs attribute)
(CategoricalLogits attribute)
(CategoricalProbs attribute)
(Cauchy attribute)
(Chi2 attribute)
(Delta attribute)
(Dirichlet attribute)
(DirichletMultinomial attribute)
(Distribution attribute)
(ExpandedDistribution attribute)
(Exponential attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(GeometricLogits attribute)
(GeometricProbs attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(ImproperUniform attribute)
(Independent attribute)
(InverseGamma attribute)
(Laplace attribute)
(LeftTruncatedDistribution attribute)
(LKJ attribute)
(LKJCholesky attribute)
(Logistic attribute)
(LogNormal attribute)
(LowRankMultivariateNormal attribute)
(MaskedDistribution attribute)
(MultinomialLogits attribute)
(MultinomialProbs attribute)
(MultivariateNormal attribute)
(NegativeBinomial2 attribute)
(NegativeBinomialLogits attribute)
(NegativeBinomialProbs attribute)
(Normal attribute)
(OrderedLogistic attribute)
(Pareto attribute)
(Poisson attribute)
(ProjectedNormal attribute)
(RightTruncatedDistribution attribute)
(SineBivariateVonMises attribute)
(SineSkewed attribute)
(SoftLaplace attribute)
(StudentT attribute)
(TransformedDistribution attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(TruncatedPolyaGamma attribute)
(TwoSidedTruncatedDistribution attribute)
(Uniform attribute)
(Unit attribute)
(VonMises attribute)
(Weibull attribute)
(ZeroInflatedPoisson attribute)
AutoBNAFNormal (class in numpyro.infer.autoguide)
AutoContinuous (class in numpyro.infer.autoguide)
autocorrelation() (in module numpyro.diagnostics)
autocovariance() (in module numpyro.diagnostics)
AutoDAIS (class in numpyro.infer.autoguide)
AutoDelta (class in numpyro.infer.autoguide)
AutoDiagonalNormal (class in numpyro.infer.autoguide)
AutoGuide (class in numpyro.infer.autoguide)
AutoIAFNormal (class in numpyro.infer.autoguide)
AutoLaplaceApproximation (class in numpyro.infer.autoguide)
AutoLowRankMultivariateNormal (class in numpyro.infer.autoguide)
AutoMultivariateNormal (class in numpyro.infer.autoguide)
AutoNormal (class in numpyro.infer.autoguide)
B
BarkerMH (class in numpyro.infer.barker)
BarkerMHState (in module numpyro.infer.barker)
batch_shape (Distribution property)
Bernoulli() (in module numpyro.distributions.discrete)
BernoulliLogits (class in numpyro.distributions.discrete)
BernoulliProbs (class in numpyro.distributions.discrete)
Beta (class in numpyro.distributions.continuous)
BetaBinomial (class in numpyro.distributions.conjugate)
BetaProportion (class in numpyro.distributions.continuous)
biject_to() (in module numpyro.distributions.transforms)
BijectorConstraint (class in numpyro.contrib.tfp.distributions)
BijectorTransform (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)
boolean (in module numpyro.distributions.constraints)
C
call_with_intermediates() (BlockNeuralAutoregressiveTransform method)
(ComposeTransform method)
(InverseAutoregressiveTransform method)
(Transform method)
can_infer_discrete (ELBO attribute)
(TraceGraph_ELBO attribute)
Categorical() (in module numpyro.distributions.discrete)
CategoricalLogits (class in numpyro.distributions.discrete)
CategoricalProbs (class in numpyro.distributions.discrete)
Cauchy (class in numpyro.distributions.continuous)
cdf() (Beta method)
(Cauchy method)
(Distribution method)
(Exponential method)
(GammaPoisson method)
(Gumbel method)
(HalfCauchy method)
(HalfNormal method)
(Laplace method)
(Logistic method)
(MixtureSameFamily method)
(Normal method)
(Pareto method)
(Poisson method)
(SoftLaplace method)
(StudentT method)
(Uniform method)
(Weibull method)
check() (Constraint method)
Chi2 (class in numpyro.distributions.continuous)
CholeskyTransform (class in numpyro.distributions.transforms)
circular (in module numpyro.distributions.constraints)
CircularReparam (class in numpyro.infer.reparam)
ClippedAdam (class in numpyro.optim)
codomain (AbsTransform attribute)
(AffineTransform property)
(BlockNeuralAutoregressiveTransform attribute)
(CholeskyTransform attribute)
(ComposeTransform property)
(CorrCholeskyTransform attribute)
(CorrMatrixCholeskyTransform attribute)
(ExpTransform property)
(InvCholeskyTransform property)
(InverseAutoregressiveTransform attribute)
(L1BallTransform attribute)
(LowerCholeskyAffine attribute)
(LowerCholeskyTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(PowerTransform attribute)
(ScaledUnitLowerCholeskyTransform attribute)
(SigmoidTransform attribute)
(SimplexToOrderedTransform attribute)
(SoftplusLowerCholeskyTransform attribute)
(SoftplusTransform attribute)
(StickBreakingTransform attribute)
(Transform attribute)
collapse (class in numpyro.handlers)
component_distribution (MixtureSameFamily property)
ComposeTransform (class in numpyro.distributions.transforms)
cond() (in module numpyro.contrib.control_flow)
condition (class in numpyro.handlers)
config_enumerate() (in module numpyro.contrib.funsor.infer_util)
consensus() (in module numpyro.infer.hmc_util)
constrain_fn() (in module numpyro.infer.util)
Constraint (class in numpyro.distributions.constraints)
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 method)
(MultivariateNormal method)
D
default_fields (HMC property)
(MCMCKernel property)
(SA property)
Delta (class in numpyro.distributions.distribution)
dependent (in module numpyro.distributions.constraints)
deterministic() (in module numpyro.primitives)
diagnostics() (NestedSampler method)
Dirichlet (class in numpyro.distributions.continuous)
DirichletMultinomial (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 property)
(CorrCholeskyTransform attribute)
(CorrMatrixCholeskyTransform attribute)
(InverseAutoregressiveTransform attribute)
(L1BallTransform attribute)
(LowerCholeskyAffine attribute)
(LowerCholeskyTransform attribute)
(OrderedTransform attribute)
(PermuteTransform attribute)
(PowerTransform attribute)
(ScaledUnitLowerCholeskyTransform attribute)
(SimplexToOrderedTransform attribute)
(SoftplusLowerCholeskyTransform attribute)
(SoftplusTransform attribute)
(StickBreakingTransform attribute)
(Transform attribute)
E
effective_sample_size() (in module numpyro.diagnostics)
ELBO (class in numpyro.infer.elbo)
enable_validation() (in module numpyro.distributions.distribution)
enable_x64() (in module numpyro.util)
entropy() (LowRankMultivariateNormal method)
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 property)
(Transform property)
event_shape (Distribution property)
expand() (Distribution method)
(Independent method)
expand_by() (Distribution method)
ExpandedDistribution (class in numpyro.distributions.distribution)
Exponential (class in numpyro.distributions.continuous)
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)
flax_module() (in module numpyro.contrib.module)
FoldedDistribution (class in numpyro.distributions.distribution)
fori_collect() (in module numpyro.util)
format_shapes() (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.distributions.continuous)
GammaPoisson (class in numpyro.distributions.conjugate)
GaussianRandomWalk (class in numpyro.distributions.continuous)
gelman_rubin() (in module numpyro.diagnostics)
Geometric() (in module numpyro.distributions.discrete)
GeometricLogits (class in numpyro.distributions.discrete)
GeometricProbs (class in numpyro.distributions.discrete)
get_base_dist() (AutoBNAFNormal method)
(AutoContinuous method)
(AutoDiagonalNormal method)
(AutoIAFNormal method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
get_diagnostics_str() (BarkerMH method)
(HMC method)
(HMCGibbs method)
(MCMCKernel method)
(SA method)
get_extra_fields() (MCMC method)
get_mask() (in module numpyro.primitives)
get_model_relations() (in module numpyro.contrib.render)
get_params() (Adagrad method)
(Adam method)
(ClippedAdam method)
(Minimize method)
(Momentum method)
(RMSProp method)
(RMSPropMomentum method)
(SGD method)
(SM3 method)
(SVI method)
get_posterior() (AutoContinuous method)
(AutoDiagonalNormal method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
get_samples() (MCMC method)
(NestedSampler method)
get_trace() (trace method)
get_transform() (AutoContinuous method)
(AutoDiagonalNormal method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
get_weighted_samples() (NestedSampler method)
greater_than() (in module numpyro.distributions.constraints)
Gumbel (class in numpyro.distributions.continuous)
H
haiku_module() (in module numpyro.contrib.module)
HalfCauchy (class in numpyro.distributions.continuous)
HalfNormal (class in numpyro.distributions.continuous)
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 property)
(Independent property)
(MaskedDistribution property)
has_rsample (Distribution property)
(ExpandedDistribution property)
(Independent property)
(MaskedDistribution property)
(TransformedDistribution property)
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)
hpdi() (in module numpyro.diagnostics)
I
icdf() (Cauchy method)
(Distribution method)
(Exponential method)
(Gumbel method)
(HalfCauchy method)
(HalfNormal method)
(Laplace method)
(Logistic method)
(Normal method)
(Pareto method)
(SoftLaplace method)
(StudentT method)
(Uniform method)
IdentityTransform (class in numpyro.distributions.transforms)
ImproperUniform (class in numpyro.distributions.distribution)
Independent (class in numpyro.distributions.distribution)
infer_config (class in numpyro.contrib.funsor.enum_messenger)
(class in numpyro.handlers)
infer_discrete() (in module numpyro.contrib.funsor.discrete)
infer_shapes() (Dirichlet static method)
(DirichletMultinomial static method)
(Distribution class method)
(LowRankMultivariateNormal static method)
(MultinomialLogits static method)
(MultinomialProbs static method)
(MultivariateNormal static method)
(OrderedLogistic static method)
(ProjectedNormal static method)
(Uniform static 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 property)
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.distributions.continuous)
is_discrete (Constraint attribute)
(Distribution property)
(MixtureSameFamily property)
L
l1_ball() (in module numpyro.distributions.constraints)
L1BallTransform (class in numpyro.distributions.transforms)
Laplace (class in numpyro.distributions.continuous)
last_state (MCMC property)
LeftTruncatedDistribution (class in numpyro.distributions.truncated)
less_than() (in module numpyro.distributions.constraints)
lift (class in numpyro.handlers)
LKJ (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)
(L1BallTransform method)
(LowerCholeskyAffine method)
(LowerCholeskyTransform method)
(OrderedTransform method)
(PermuteTransform method)
(PowerTransform method)
(ScaledUnitLowerCholeskyTransform method)
(SigmoidTransform method)
(SimplexToOrderedTransform method)
(SoftplusLowerCholeskyTransform method)
(SoftplusTransform method)
(StickBreakingTransform method)
(Transform method)
log_density() (in module numpyro.contrib.funsor.infer_util)
(in module numpyro.infer.util)
log_likelihood() (in module numpyro.infer.util)
log_prob() (BernoulliLogits method)
(BernoulliProbs method)
(Beta method)
(BetaBinomial method)
(BinomialLogits method)
(BinomialProbs method)
(CategoricalLogits method)
(CategoricalProbs method)
(Cauchy method)
(Delta method)
(Dirichlet method)
(DirichletMultinomial method)
(Distribution method)
(ExpandedDistribution method)
(Exponential method)
(FoldedDistribution method)
(Gamma method)
(GammaPoisson method)
(GaussianRandomWalk method)
(GeometricLogits method)
(GeometricProbs method)
(Gumbel method)
(HalfCauchy method)
(HalfNormal method)
(ImproperUniform method)
(Independent method)
(Laplace method)
(LeftTruncatedDistribution method)
(LKJCholesky method)
(Logistic method)
(LowRankMultivariateNormal method)
(MaskedDistribution method)
(MixtureSameFamily method)
(MultinomialLogits method)
(MultinomialProbs method)
(MultivariateNormal method)
(NegativeBinomialLogits method)
(Normal method)
(Poisson method)
(ProjectedNormal method)
(RightTruncatedDistribution method)
(SineBivariateVonMises method)
(SineSkewed method)
(SoftLaplace method)
(StudentT method)
(TransformedDistribution method)
(TruncatedPolyaGamma method)
(TwoSidedTruncatedDistribution method)
(Uniform method)
(Unit method)
(VonMises method)
(Weibull method)
Logistic (class in numpyro.distributions.continuous)
logits() (BernoulliProbs method)
(BinomialProbs method)
(CategoricalProbs method)
(GeometricProbs method)
(MultinomialProbs method)
LogNormal (class in numpyro.distributions.continuous)
loss() (ELBO method)
(RenyiELBO method)
(TraceGraph_ELBO method)
loss_with_mutable_state() (ELBO method)
(Trace_ELBO method)
(TraceMeanField_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)
max_sample_iter (SineBivariateVonMises attribute)
MCMC (class in numpyro.infer.mcmc)
MCMCKernel (class in numpyro.infer.mcmc)
mean (BernoulliLogits property)
(BernoulliProbs property)
(Beta property)
(BetaBinomial property)
(BinomialLogits property)
(BinomialProbs property)
(CategoricalLogits property)
(CategoricalProbs property)
(Cauchy property)
(Delta property)
(Dirichlet property)
(DirichletMultinomial property)
(Distribution property)
(ExpandedDistribution property)
(Exponential property)
(Gamma property)
(GammaPoisson property)
(GaussianRandomWalk property)
(GeometricLogits property)
(GeometricProbs property)
(Gumbel property)
(HalfCauchy property)
(HalfNormal property)
(Independent property)
(InverseGamma property)
(Laplace property)
(LKJ property)
(Logistic property)
(LogNormal property)
(LowRankMultivariateNormal property)
(MaskedDistribution property)
(MixtureSameFamily property)
(MultinomialLogits property)
(MultinomialProbs property)
(MultivariateNormal property)
(Normal property)
(Pareto property)
(Poisson property)
(ProjectedNormal property)
(SineBivariateVonMises property)
(SineSkewed property)
(SoftLaplace property)
(StudentT property)
(TransformedDistribution property)
(TruncatedCauchy property)
(TruncatedNormal property)
(Uniform property)
(VonMises property)
(Weibull property)
median() (AutoDelta method)
(AutoDiagonalNormal method)
(AutoGuide method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
(AutoNormal method)
MetropolisAdjustedLangevinAlgorithm (class in numpyro.contrib.tfp.mcmc)
Minimize (class in numpyro.optim)
MixedHMC (class in numpyro.infer.mixed_hmc)
mixing_distribution (MixtureSameFamily property)
mixture_dim (MixtureSameFamily property)
mixture_size (MixtureSameFamily property)
MixtureSameFamily (class in numpyro.distributions.mixtures)
mode (ProjectedNormal property)
model (BarkerMH property)
(HMC property)
(HMCGibbs property)
(SA property)
module
numpyro.contrib.funsor
numpyro.contrib.indexing
numpyro.contrib.tfp.distributions
numpyro.contrib.tfp.mcmc
numpyro.diagnostics
numpyro.handlers
numpyro.infer.autoguide
numpyro.infer.reparam
numpyro.infer.util
numpyro.optim
numpyro.primitives
numpyro.util
module() (in module numpyro.primitives)
Momentum (class in numpyro.optim)
multinomial() (in module numpyro.distributions.constraints)
Multinomial() (in module numpyro.distributions.discrete)
MultinomialLogits (class in numpyro.distributions.discrete)
MultinomialProbs (class in numpyro.distributions.discrete)
MultivariateNormal (class in numpyro.distributions.continuous)
N
NegativeBinomial() (in module numpyro.distributions.conjugate)
NegativeBinomial2 (class in numpyro.distributions.conjugate)
NegativeBinomialLogits (class in numpyro.distributions.conjugate)
NegativeBinomialProbs (class in numpyro.distributions.conjugate)
NestedSampler (class in numpyro.contrib.nested_sampling)
NeuTraReparam (class in numpyro.infer.reparam)
nonnegative_integer (in module numpyro.distributions.constraints)
norm_const() (SineBivariateVonMises method)
Normal (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
optax_to_numpyro() (in module numpyro.contrib.optim)
ordered_vector (in module numpyro.distributions.constraints)
OrderedLogistic (class in numpyro.distributions.discrete)
OrderedTransform (class in numpyro.distributions.transforms)
P
param() (in module numpyro.primitives)
parametric() (in module numpyro.infer.hmc_util)
parametric_draws() (in module numpyro.infer.hmc_util)
Pareto (class in numpyro.distributions.continuous)
PermuteTransform (class in numpyro.distributions.transforms)
plate (class in numpyro.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.distributions.discrete)
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 property)
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)
PowerTransform (class in numpyro.distributions.transforms)
precision_matrix() (LowRankMultivariateNormal method)
(MultivariateNormal method)
Predictive (class in numpyro.infer.util)
print_summary() (in module numpyro.diagnostics)
(MCMC method)
(NestedSampler method)
prng_key() (in module numpyro.primitives)
PRNGIdentity (class in numpyro.distributions.discrete)
probs() (BernoulliLogits method)
(BinomialLogits method)
(CategoricalLogits method)
(GeometricLogits method)
(MultinomialLogits method)
process_message() (block method)
(collapse method)
(condition method)
(do method)
(enum method)
(infer_config method)
,
[1]
(lift method)
(mask method)
(plate method)
(reparam method)
(replay method)
(scale method)
(scope method)
(seed method)
(substitute method)
ProjectedNormal (class in numpyro.distributions.directional)
ProjectedNormalReparam (class in numpyro.infer.reparam)
Q
quantiles() (AutoDiagonalNormal method)
(AutoGuide method)
(AutoLaplaceApproximation method)
(AutoLowRankMultivariateNormal method)
(AutoMultivariateNormal method)
(AutoNormal method)
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)
render_model() (in module numpyro.contrib.render)
RenyiELBO (class in numpyro.infer.elbo)
reparam (class in numpyro.handlers)
Reparam (class in numpyro.infer.reparam)
reparam() (NeuTraReparam method)
reparameterized_params (Independent property)
reparametrized_params (Beta attribute)
(BetaProportion attribute)
(Cauchy attribute)
(Chi2 attribute)
(Delta attribute)
(Dirichlet attribute)
(Distribution attribute)
(Exponential attribute)
(Gamma attribute)
(GaussianRandomWalk attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(InverseGamma attribute)
(Laplace attribute)
(LeftTruncatedDistribution attribute)
(LKJ attribute)
(LKJCholesky attribute)
(Logistic attribute)
(LogNormal attribute)
(LowRankMultivariateNormal attribute)
(MultivariateNormal attribute)
(Normal attribute)
(Pareto attribute)
(ProjectedNormal attribute)
(RightTruncatedDistribution attribute)
(SoftLaplace attribute)
(StudentT attribute)
(TruncatedCauchy attribute)
(TruncatedNormal attribute)
(TwoSidedTruncatedDistribution attribute)
(Uniform attribute)
(VonMises attribute)
(Weibull attribute)
replay (class in numpyro.handlers)
ReplicaExchangeMC (class in numpyro.contrib.tfp.mcmc)
RightTruncatedDistribution (class in numpyro.distributions.truncated)
RMSProp (class in numpyro.optim)
RMSPropMomentum (class in numpyro.optim)
rsample() (Distribution method)
(ExpandedDistribution method)
(Independent method)
(MaskedDistribution method)
(TransformedDistribution method)
run() (MCMC method)
(NestedSampler method)
(SVI method)
S
SA (class in numpyro.infer.sa)
sample() (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)
(HalfCauchy method)
(HalfNormal method)
(HMC method)
(HMCECS method)
(HMCGibbs method)
(in module numpyro.primitives)
(Independent method)
(Laplace method)
(LeftTruncatedDistribution method)
(LKJCholesky method)
(Logistic method)
(LowRankMultivariateNormal method)
(MaskedDistribution method)
(MCMCKernel method)
(MixedHMC method)
(MixtureSameFamily method)
(MultinomialLogits method)
(MultinomialProbs method)
(MultivariateNormal method)
(Normal method)
(Poisson method)
(PRNGIdentity method)
(ProjectedNormal method)
(RightTruncatedDistribution method)
(SA method)
(SineBivariateVonMises method)
(SineSkewed method)
(SoftLaplace method)
(StudentT method)
(TransformedDistribution method)
(TruncatedPolyaGamma method)
(TwoSidedTruncatedDistribution method)
(Uniform method)
(Unit method)
(VonMises method)
(Weibull method)
sample_field (BarkerMH property)
(HMC property)
(HMCGibbs attribute)
(MCMCKernel property)
(SA property)
sample_kernel() (in module numpyro.infer.hmc.hmc)
sample_posterior() (AutoContinuous method)
(AutoDAIS method)
(AutoDelta method)
(AutoGuide method)
(AutoLaplaceApproximation method)
(AutoNormal method)
sample_with_intermediates() (Distribution method)
(ExpandedDistribution method)
(MixtureSameFamily 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 method)
scale_tril_constraint (AutoMultivariateNormal attribute)
scaled_unit_lower_cholesky (in module numpyro.distributions.constraints)
ScaledUnitLowerCholeskyTransform (class in numpyro.distributions.transforms)
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)
SimplexToOrderedTransform (class in numpyro.distributions.transforms)
SineBivariateVonMises (class in numpyro.distributions.directional)
SineSkewed (class in numpyro.distributions.directional)
SliceSampler (class in numpyro.contrib.tfp.mcmc)
SM3 (class in numpyro.optim)
SoftLaplace (class in numpyro.distributions.continuous)
softplus_lower_cholesky (in module numpyro.distributions.constraints)
softplus_positive (in module numpyro.distributions.constraints)
SoftplusLowerCholeskyTransform (class in numpyro.distributions.transforms)
SoftplusTransform (class in numpyro.distributions.transforms)
sphere (in module numpyro.distributions.constraints)
split_gelman_rubin() (in module numpyro.diagnostics)
stable_update() (SVI method)
StickBreakingTransform (class in numpyro.distributions.transforms)
StudentT (class in numpyro.distributions.continuous)
subsample() (in module numpyro.primitives)
substitute (class in numpyro.handlers)
summary() (in module numpyro.diagnostics)
support (BernoulliLogits attribute)
(BernoulliProbs attribute)
(Beta attribute)
(BetaBinomial property)
(BetaProportion attribute)
(BinomialLogits property)
(BinomialProbs property)
(CategoricalLogits property)
(CategoricalProbs property)
(Cauchy attribute)
(Delta property)
(Dirichlet attribute)
(DirichletMultinomial property)
(Distribution attribute)
(ExpandedDistribution property)
(Exponential attribute)
(FoldedDistribution attribute)
(Gamma attribute)
(GammaPoisson attribute)
(GaussianRandomWalk attribute)
(GeometricLogits attribute)
(GeometricProbs attribute)
(Gumbel attribute)
(HalfCauchy attribute)
(HalfNormal attribute)
(ImproperUniform attribute)
(Independent property)
(InverseGamma attribute)
(Laplace attribute)
(LeftTruncatedDistribution property)
(LKJ attribute)
(LKJCholesky attribute)
(Logistic attribute)
(LogNormal attribute)
(LowRankMultivariateNormal attribute)
(MaskedDistribution property)
(MixtureSameFamily property)
(MultinomialLogits property)
(MultinomialProbs property)
(MultivariateNormal attribute)
(NegativeBinomial2 attribute)
(NegativeBinomialLogits attribute)
(NegativeBinomialProbs attribute)
(Normal attribute)
(Pareto property)
(Poisson attribute)
(ProjectedNormal attribute)
(RightTruncatedDistribution property)
(SineBivariateVonMises attribute)
(SineSkewed attribute)
(SoftLaplace attribute)
(StudentT attribute)
(TransformedDistribution property)
(TruncatedPolyaGamma attribute)
(TwoSidedTruncatedDistribution property)
(Uniform property)
(Unit attribute)
(VonMises attribute)
(Weibull attribute)
(ZeroInflatedPoisson attribute)
supported_types (LeftTruncatedDistribution attribute)
(RightTruncatedDistribution attribute)
(TwoSidedTruncatedDistribution attribute)
SVI (class in numpyro.infer.svi)
T
taylor_proxy() (HMCECS static method)
(in module numpyro.infer.hmc_gibbs)
TFPDistribution (class in numpyro.contrib.tfp.distributions)
TFPKernel (class in numpyro.contrib.tfp.mcmc)
to_data() (in module numpyro.contrib.funsor.enum_messenger)
to_event() (Distribution method)
to_funsor() (in module numpyro.contrib.funsor.enum_messenger)
trace (class in numpyro.contrib.funsor.enum_messenger)
(class in numpyro.handlers)
Trace_ELBO (class in numpyro.infer.elbo)
TraceGraph_ELBO (class in numpyro.infer.elbo)
TraceMeanField_ELBO (class in numpyro.infer.elbo)
Transform (class in numpyro.distributions.transforms)
transform_fn() (in module numpyro.infer.util)
transform_sample() (NeuTraReparam method)
TransformedDistribution (class in numpyro.distributions.distribution)
TransformReparam (class in numpyro.infer.reparam)
tree_flatten() (Delta method)
(Distribution method)
(ExpandedDistribution method)
(FoldedDistribution method)
(GaussianRandomWalk method)
(ImproperUniform method)
(Independent method)
(InverseGamma method)
(LeftTruncatedDistribution method)
(LKJ method)
(LKJCholesky method)
(LogNormal method)
(MaskedDistribution method)
(MixtureSameFamily method)
(MultivariateNormal method)
(Pareto method)
(RightTruncatedDistribution method)
(TransformedDistribution method)
(TruncatedCauchy method)
(TruncatedNormal method)
(TruncatedPolyaGamma method)
(TwoSidedTruncatedDistribution method)
(Uniform method)
tree_unflatten() (Delta class method)
(Distribution class method)
(ExpandedDistribution class method)
(FoldedDistribution class method)
(GaussianRandomWalk class method)
(Independent class method)
(LeftTruncatedDistribution class method)
(LKJ class method)
(LKJCholesky class method)
(MaskedDistribution class method)
(MixtureSameFamily class method)
(MultivariateNormal class method)
(RightTruncatedDistribution class method)
(TruncatedCauchy class method)
(TruncatedNormal class method)
(TruncatedPolyaGamma class method)
(TwoSidedTruncatedDistribution class method)
(Uniform class method)
TruncatedCauchy (class in numpyro.distributions.truncated)
TruncatedDistribution() (in module numpyro.distributions.truncated)
TruncatedNormal (class in numpyro.distributions.truncated)
TruncatedPolyaGamma (class in numpyro.distributions.truncated)
truncation_point (TruncatedPolyaGamma attribute)
TwoSidedTruncatedDistribution (class in numpyro.distributions.truncated)
U
UncalibratedHamiltonianMonteCarlo (class in numpyro.contrib.tfp.mcmc)
UncalibratedLangevin (class in numpyro.contrib.tfp.mcmc)
UncalibratedRandomWalk (class in numpyro.contrib.tfp.mcmc)
Uniform (class in numpyro.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 property)
(BernoulliProbs property)
(Beta property)
(BetaBinomial property)
(BinomialLogits property)
(BinomialProbs property)
(CategoricalLogits property)
(CategoricalProbs property)
(Cauchy property)
(Delta property)
(Dirichlet property)
(DirichletMultinomial property)
(Distribution property)
(ExpandedDistribution property)
(Exponential property)
(Gamma property)
(GammaPoisson property)
(GaussianRandomWalk property)
(GeometricLogits property)
(GeometricProbs property)
(Gumbel property)
(HalfCauchy property)
(HalfNormal property)
(Independent property)
(InverseGamma property)
(Laplace property)
(Logistic property)
(LogNormal property)
(MaskedDistribution property)
(MixtureSameFamily property)
(MultinomialLogits property)
(MultinomialProbs property)
(MultivariateNormal property)
(Normal property)
(Pareto property)
(Poisson property)
(SoftLaplace property)
(StudentT property)
(TransformedDistribution property)
(TruncatedCauchy property)
(TruncatedNormal property)
(Uniform property)
(VonMises property)
(Weibull property)
variance() (LowRankMultivariateNormal method)
Vindex (class in numpyro.contrib.indexing)
vindex() (in module numpyro.contrib.indexing)
VonMises (class in numpyro.distributions.directional)
W
warmup() (MCMC method)
Weibull (class in numpyro.distributions.continuous)
Z
ZeroInflatedDistribution() (in module numpyro.distributions.discrete)
ZeroInflatedNegativeBinomial2() (in module numpyro.distributions.conjugate)
ZeroInflatedPoisson (class in numpyro.distributions.discrete)
Read the Docs
v: 0.8.0
Versions
latest
stable
0.8.0
0.7.2
0.7.1
0.7.0
0.6.0
0.5.0
0.4.1
0.4.0
0.3.0
v0.2.4
v0.2.3
v0.2.2
v0.2.1
v0.2.0
v0.1.0
fix-docs
Downloads
On Read the Docs
Project Home
Builds