Index

_

__add__() (AffineNormal method) (Gaussian method) __call__() (Adam method) (CoalescentRateLikelihood method) (Distribution method) (EnergyDistance method) (Forecaster method) (GuideMessenger method) (HMCForecaster method) (HorovodOptimizer method) (JitTrace_ELBO method) (Messenger method) (PyroLRScheduler method) (PyroOptim method) (Reparam method) (Strategy method) (TorchDistributionMixin method)

__getitem__() (AffineNormal method) (Gaussian method) _CorrMatrix (class in pyro.distributions.constraints) _Integer (class in pyro.distributions.constraints) _OrderedVector (class in pyro.distributions.constraints) _PositiveOrderedVector (class in pyro.distributions.constraints) _SoftplusLowerCholesky (class in pyro.distributions.constraints) _SoftplusPositive (class in pyro.distributions.constraints) _Sphere (class in pyro.distributions.constraints) _UnitLowerCholesky (class in pyro.distributions.constraints)

A

Adadelta() (in module pyro.optim.pytorch_optimizers) Adagrad() (in module pyro.optim.pytorch_optimizers) AdagradRMSProp (class in pyro.optim.adagrad_rmsprop) AdagradRMSProp() (in module pyro.optim.optim) Adam (class in pyro.contrib.minipyro) Adam() (in module pyro.optim.pytorch_optimizers) Adamax() (in module pyro.optim.pytorch_optimizers) AdamW() (in module pyro.optim.pytorch_optimizers) add() (AutoGuideList method) (List method) (LSH method) add_edge() (Trace method) add_module() (PyroModule method) add_node() (Trace method) affine_autoregressive() (in module pyro.distributions.transforms) affine_coupling() (in module pyro.distributions.transforms) AffineAutoregressive (class in pyro.distributions.transforms) AffineBeta (class in pyro.distributions) AffineCoupling (class in pyro.distributions.transforms) AffineNormal (class in pyro.ops.gaussian) all_escape() (in module pyro.poutine.util) allocate() (DimStack method) am_i_wrapped() (in module pyro.poutine.runtime) append() (AutoGuideList method) (FullyConnected method) apply() (ConjugateReparam method) (GumbelSoftmaxReparam method) (LatentStableReparam method) (LinearHMMReparam method) (LocScaleReparam method) (NeuTraReparam method) (ProjectedNormalReparam method) (Reparam method) (SplitReparam method) (StableReparam method) (StructuredReparam method) (StudentTReparam method) (SymmetricStableReparam method) (TransformReparam method) (UnitJacobianReparam method) apply_optimizer() (in module pyro.contrib.funsor.infer.traceenum_elbo) apply_stack() (in module pyro.contrib.minipyro) (in module pyro.poutine.runtime) approx_log_prob_tol (BetaBinomial attribute) ApproxSet (class in pyro.contrib.tracking.hashing) arg_constraints (AffineBeta attribute) (AsymmetricLaplace attribute) (AVFMultivariateNormal attribute) (BetaBinomial attribute) (CoalescentTimes attribute) (CoalescentTimesWithRate attribute) (Delta attribute) (DirichletMultinomial attribute) (DiscreteHMM attribute) (Empirical attribute) (ExtendedBetaBinomial attribute) (ExtendedBinomial attribute) (GammaGaussianHMM attribute) (GammaPoisson attribute) (GaussianHMM attribute) (GaussianMRF attribute) (GaussianScaleMixture attribute) (GroupedNormalNormal attribute) (ImproperUniform attribute) (IndependentHMM attribute) (InverseGamma attribute) (LinearHMM attribute) (LKJ attribute) (Logistic attribute) (LogNormalNegativeBinomial attribute) (MaskedDistribution attribute) (MaskedMixture attribute) (MixtureOfDiagNormals attribute) (MixtureOfDiagNormalsSharedCovariance attribute) (MultivariateStudentT attribute) (OMTMultivariateNormal attribute) (OneOneMatching attribute) (OneTwoMatching attribute) (OrderedLogistic attribute) (ProjectedNormal attribute) (Rejector attribute) (SineBivariateVonMises attribute) (SineSkewed attribute) (SkewLogistic attribute) (SoftAsymmetricLaplace attribute) (SoftLaplace attribute) (SpanningTree attribute) (Stable attribute) (TruncatedPolyaGamma attribute) (Unit attribute) (VonMises3D attribute) (ZeroInflatedDistribution attribute) (ZeroInflatedNegativeBinomial attribute) (ZeroInflatedPoisson attribute)

args (Message attribute) as_complex() (in module pyro.ops.tensor_utils) ASGD() (in module pyro.optim.pytorch_optimizers) AsymmetricLaplace (class in pyro.distributions) auto_goodness_of_fit() (in module pyro.distributions.testing.gof) AutoCallable (class in pyro.infer.autoguide) AutoContinuous (class in pyro.infer.autoguide) autocorrelation() (in module pyro.ops.stats) autocovariance() (in module pyro.ops.stats) AutoDelta (class in pyro.infer.autoguide) AutoDiagonalNormal (class in pyro.infer.autoguide) AutoDiscreteParallel (class in pyro.infer.autoguide) AutoGaussian (class in pyro.infer.autoguide) AutoGuide (class in pyro.infer.autoguide) autoguide() (Parameterized method), [1] AutoGuideList (class in pyro.infer.autoguide) AutoHierarchicalNormalMessenger (class in pyro.infer.autoguide) AutoIAFNormal (class in pyro.infer.autoguide) AutoLaplaceApproximation (class in pyro.infer.autoguide) AutoLowRankMultivariateNormal (class in pyro.infer.autoguide) AutoMessenger (class in pyro.infer.autoguide) AutoMultivariateNormal (class in pyro.infer.autoguide) autoname() (in module pyro.contrib.autoname) AutoNormal (class in pyro.infer.autoguide) AutoNormalizingFlow (class in pyro.infer.autoguide) AutoNormalMessenger (class in pyro.infer.autoguide) autoregressive (AffineAutoregressive attribute) (BlockAutoregressive attribute) (NeuralAutoregressive attribute) (Polynomial attribute) (SplineAutoregressive attribute) AutoRegressiveMessenger (class in pyro.infer.autoguide) AutoRegressiveNN (class in pyro.nn.auto_reg_nn) AutoReparam (class in pyro.infer.reparam.strategies) AutoStructured (class in pyro.infer.autoguide) AVFMultivariateNormal (class in pyro.distributions)

B

backtest() (in module pyro.contrib.forecast.evaluate) BackwardSampleMessenger (class in pyro.infer.traceenum_elbo) bandwidth_factor (IMQSteinKernel property) (RBFSteinKernel property) barrier() (in module pyro.primitives) batch_shape (AffineNormal property) (Gaussian property) (TorchDistributionMixin property) BatchNorm (class in pyro.distributions.transforms) batchnorm() (in module pyro.distributions.transforms) Bernoulli (class in pyro.distributions) BernoulliNet (class in pyro.contrib.cevae) Beta (class in pyro.distributions) beta_binomial_dist() (in module pyro.contrib.epidemiology.distributions) BetaBinomial (class in pyro.distributions) bijective (AffineAutoregressive attribute) (AffineCoupling attribute) (BatchNorm attribute) (BlockAutoregressive attribute) (CholeskyTransform attribute) (ConditionalAffineAutoregressive attribute) (ConditionalAffineCoupling attribute) (ConditionalGeneralizedChannelPermute attribute) (ConditionalHouseholder attribute) (ConditionalMatrixExponential attribute) (ConditionalNeuralAutoregressive attribute) (ConditionalPlanar attribute) (ConditionalRadial attribute) (ConditionalSpline attribute) (ConditionalSplineAutoregressive attribute) (CorrMatrixCholeskyTransform attribute) (DiscreteCosineTransform attribute) (ELUTransform attribute) (GeneralizedChannelPermute attribute) (HaarTransform attribute) (Householder attribute) (LeakyReLUTransform attribute) (LowerCholeskyAffine attribute) (MatrixExponential attribute) (NeuralAutoregressive attribute) (Normalize attribute) (OrderedTransform attribute) (Permute attribute) (Planar attribute) (Polynomial attribute) (PositivePowerTransform attribute) (Radial attribute) (SoftplusTransform attribute) (Spline attribute) (SplineAutoregressive attribute) (SplineCoupling attribute) (Sylvester attribute)

Binary (class in pyro.contrib.gp.likelihoods) Binomial (class in pyro.distributions) binomial_dist() (in module pyro.contrib.epidemiology.distributions) bio_phylo_to_times() (in module pyro.distributions.coalescent) BiosequenceDataset (class in pyro.contrib.mue.dataloaders) block (class in pyro.contrib.minipyro) block() (in module pyro.poutine.handlers) block_autoregressive() (in module pyro.distributions.transforms) block_diag_embed() (in module pyro.ops.tensor_utils) block_diagonal() (in module pyro.ops.tensor_utils) block_messengers() (in module pyro.poutine.messenger) block_plate() (in module pyro.poutine.plate_messenger) BlockAutoregressive (class in pyro.distributions.transforms) BlockMassMatrix (class in pyro.infer.mcmc) BlockMessenger (class in pyro.poutine.block_messenger) broadcast() (in module pyro.poutine.handlers) BroadcastMessenger (class in pyro.poutine.broadcast_messenger) Brownian (class in pyro.contrib.gp.kernels)

C

call() (AutoGuide method) (AutoMessenger method) (Predictive method) (WeighedPredictive method) cat() (Gaussian static method) Categorical (class in pyro.distributions) Cauchy (class in pyro.distributions) cdf() (Logistic method) (SkewLogistic method) (SoftLaplace method) CEVAE (class in pyro.contrib.cevae) ChainedScheduler() (in module pyro.optim.pytorch_optimizers) Chi2 (class in pyro.distributions) cholesky_solve() (in module pyro.ops.tensor_utils) CholeskyTransform (class in pyro.distributions.transforms) cleanup() (HMC method) (MCMCKernel method) clear() (in module pyro.nn.module) (ParamStoreDict method) clear_cache() (ConditionalTransformedDistribution method) (HMC method) clear_param_store() (in module pyro.contrib.funsor) (in module pyro.primitives) ClippedAdam (class in pyro.optim.clipped_adam) ClippedAdam() (in module pyro.optim.optim) CoalescentRateLikelihood (class in pyro.distributions) CoalescentTimes (class in pyro.distributions) CoalescentTimesWithRate (class in pyro.distributions) codomain (AffineAutoregressive attribute) (AffineCoupling property) (BatchNorm attribute) (BlockAutoregressive attribute) (CholeskyTransform attribute) (ConditionalAffineAutoregressive attribute) (ConditionalAffineCoupling attribute) (ConditionalGeneralizedChannelPermute attribute) (ConditionalHouseholder attribute) (ConditionalMatrixExponential attribute) (ConditionalNeuralAutoregressive attribute) (ConditionalPlanar attribute) (ConditionalRadial attribute) (ConditionalSpline attribute) (ConditionalSplineAutoregressive attribute) (CorrMatrixCholeskyTransform attribute) (DiscreteCosineTransform property) (ELUTransform attribute) (GeneralizedChannelPermute attribute) (HaarTransform property) (Householder attribute) (LeakyReLUTransform attribute) (LowerCholeskyAffine attribute) (MatrixExponential attribute) (NeuralAutoregressive attribute) (Normalize attribute) (OrderedTransform attribute) (Permute property) (Planar attribute) (Polynomial attribute) (PositivePowerTransform attribute) (Radial attribute) (SimplexToOrderedTransform attribute) (SoftplusLowerCholeskyTransform attribute) (SoftplusTransform attribute) (Spline attribute) (SplineAutoregressive attribute) (SplineCoupling attribute) (Sylvester attribute) (TransformModule attribute) (UnitLowerCholeskyTransform attribute) collapse() (in module pyro.poutine.handlers) CollapseMessenger (class in pyro.poutine.collapse_messenger) Combination (class in pyro.contrib.gp.kernels) CompartmentalModel (class in pyro.contrib.epidemiology.compartmental) ComposeTransformModule (class in pyro.distributions) compute_flows() (CompartmentalModel method) compute_log_prob() (Trace method) compute_marginals() (in module pyro.contrib.tracking.assignment) (TraceEnum_ELBO method) compute_marginals_bp() (in module pyro.contrib.tracking.assignment) compute_marginals_persistent() (in module pyro.contrib.tracking.assignment) compute_marginals_persistent_bp() (in module pyro.contrib.tracking.assignment) compute_marginals_sparse_bp() (in module pyro.contrib.tracking.assignment) compute_score_parts() (Trace method) concentration (DirichletMultinomial property) (GammaPoisson property) (InverseGamma property) concentration0 (AffineBeta property) (BetaBinomial property) concentration1 (AffineBeta property) (BetaBinomial property)

cond_indep_stack (Message attribute) CondIndepStackFrame (class in pyro.poutine.indep_messenger) condition() (AffineNormal method) (ConditionalAffineAutoregressive method) (ConditionalAffineCoupling method) (ConditionalDistribution method) (ConditionalGeneralizedChannelPermute method) (ConditionalHouseholder method) (ConditionalMatrixExponential method) (ConditionalNeuralAutoregressive method) (ConditionalPlanar method) (ConditionalRadial method) (ConditionalSpline method) (ConditionalSplineAutoregressive method) (ConditionalTransform method) (ConditionalTransformedDistribution method) (Gaussian method) (in module pyro.contrib.funsor) (in module pyro.poutine.handlers) conditional() (in module pyro.contrib.gp.util) conditional_affine_autoregressive() (in module pyro.distributions.transforms) conditional_affine_coupling() (in module pyro.distributions.transforms) conditional_generalized_channel_permute() (in module pyro.distributions.transforms) conditional_householder() (in module pyro.distributions.transforms) conditional_matrix_exponential() (in module pyro.distributions.transforms) conditional_neural_autoregressive() (in module pyro.distributions.transforms) conditional_planar() (in module pyro.distributions.transforms) conditional_radial() (in module pyro.distributions.transforms) conditional_spline() (in module pyro.distributions.transforms) conditional_spline_autoregressive() (in module pyro.distributions.transforms) ConditionalAffineAutoregressive (class in pyro.distributions.transforms) ConditionalAffineCoupling (class in pyro.distributions.transforms) ConditionalAutoRegressiveNN (class in pyro.nn.auto_reg_nn) ConditionalDenseNN (class in pyro.nn.dense_nn) ConditionalDistribution (class in pyro.distributions) ConditionalGeneralizedChannelPermute (class in pyro.distributions.transforms) ConditionalHouseholder (class in pyro.distributions.transforms) ConditionalMatrixExponential (class in pyro.distributions.transforms) ConditionalNeuralAutoregressive (class in pyro.distributions.transforms) ConditionalPlanar (class in pyro.distributions.transforms) ConditionalRadial (class in pyro.distributions.transforms) ConditionalSpline (class in pyro.distributions.transforms) ConditionalSplineAutoregressive (class in pyro.distributions.transforms) ConditionalTransform (class in pyro.distributions) ConditionalTransformedDistribution (class in pyro.distributions) ConditionalTransformModule (class in pyro.distributions) ConditionMessenger (class in pyro.poutine.condition_messenger) config_enumerate() (in module pyro.infer.enum) configure() (AutoReparam method) (BlockMassMatrix method) (MinimalReparam method) (Strategy method) conjugate_update() (Distribution method) (GaussianHMM method) (MaskedDistribution method) ConjugateReparam (class in pyro.infer.reparam.conjugate) Constant (class in pyro.contrib.gp.kernels) ConstantLR() (in module pyro.optim.pytorch_optimizers) constrained_gamma (BatchNorm property) constraint (PyroParam attribute) constraints (StateDict attribute) context() (in module pyro.settings) continuation (Message attribute) ContinuousBernoulli (class in pyro.distributions) contract() (in module pyro.ops.einsum) contract_expression() (in module pyro.ops.einsum) convolve() (in module pyro.ops.tensor_utils) copy() (Trace method) Coregionalize (class in pyro.contrib.gp.kernels) CorrMatrixCholeskyTransform (class in pyro.distributions.transforms) Cosine (class in pyro.contrib.gp.kernels) CosineAnnealingLR() (in module pyro.optim.pytorch_optimizers) CosineAnnealingWarmRestarts() (in module pyro.optim.pytorch_optimizers) counter (CondIndepStackFrame attribute) CountMeanStats (class in pyro.ops.streaming) CountMeanVarianceStats (class in pyro.ops.streaming) CountStats (class in pyro.ops.streaming) cov (EKFState property) (Measurement property) cov2pv() (DynamicModel method) (Ncp method) (Ncv method) cov_pv (EKFState property) covariance_matrix (MultivariateStudentT property) crps_empirical() (in module pyro.ops.stats) current_read_env (DimStack property) current_write_env (DimStack property) CyclicLR() (in module pyro.optim.pytorch_optimizers)

D

dct() (in module pyro.ops.tensor_utils) DCTAdam() (in module pyro.optim.optim) DEFAULT_FIRST_DIM (DimStack attribute) default_process_message() (in module pyro.poutine.runtime) Delta (class in pyro.distributions) DenseNN (class in pyro.nn.dense_nn) density_goodness_of_fit() (in module pyro.distributions.testing.gof) DependentMaternGP (class in pyro.contrib.timeseries.gp) detach_() (Trace method) deterministic() (in module pyro.contrib.funsor) (in module pyro.primitives) DiagNormalNet (class in pyro.contrib.cevae) diagnostics() (HMC method) (in module pyro.infer.mcmc.util) (MCMC method) (MCMCKernel method) (RandomWalkKernel method) (StreamingMCMC method) Dict (class in pyro.contrib.autoname.named) differentiable_loss() (ELBO method) (Jit_ELBO method) (JitTrace_ELBO method) (JitTraceEnum_ELBO method) (JitTraceMeanField_ELBO method) (Trace_ELBO method), [1] (TraceEnum_ELBO method), [1] (TraceMarkovEnum_ELBO method) (TraceTMC_ELBO method), [1] DifferentiableDynamicModel (class in pyro.contrib.tracking.dynamic_models) DifferentiableMeasurement (class in pyro.contrib.tracking.measurements) dim (CondIndepStackFrame attribute) dim() (Gaussian method) dim_type (DimRequest attribute) dimension (DynamicModel property) (EKFState property) (Measurement property) dimension_pv (DynamicModel property) (EKFState property) DimRequest (class in pyro.contrib.funsor.handlers.runtime) DimStack (class in pyro.contrib.funsor.handlers.runtime) DimType (class in pyro.contrib.funsor.handlers.runtime) Dirichlet (class in pyro.distributions) DirichletMultinomial (class in pyro.distributions) discrete_escape() (in module pyro.poutine.util) DiscreteCosineReparam (class in pyro.infer.reparam.discrete_cosine) DiscreteCosineTransform (class in pyro.distributions.transforms) DiscreteHMM (class in pyro.distributions) dist (RandomVariable property) Distribution (class in pyro.distributions) DistributionNet (class in pyro.contrib.cevae) do() (in module pyro.contrib.funsor) (in module pyro.poutine.handlers)

domain (AffineAutoregressive attribute) (AffineCoupling property) (BatchNorm attribute) (BlockAutoregressive attribute) (CholeskyTransform attribute) (ConditionalAffineAutoregressive attribute) (ConditionalAffineCoupling attribute) (ConditionalGeneralizedChannelPermute attribute) (ConditionalHouseholder attribute) (ConditionalMatrixExponential attribute) (ConditionalNeuralAutoregressive attribute) (ConditionalPlanar attribute) (ConditionalRadial attribute) (ConditionalSpline attribute) (ConditionalSplineAutoregressive attribute) (CorrMatrixCholeskyTransform attribute) (DiscreteCosineTransform property) (ELUTransform attribute) (GeneralizedChannelPermute attribute) (HaarTransform property) (Householder attribute) (LeakyReLUTransform attribute) (LowerCholeskyAffine attribute) (MatrixExponential attribute) (NeuralAutoregressive attribute) (Normalize attribute) (OrderedTransform attribute) (Permute property) (Planar attribute) (Polynomial attribute) (PositivePowerTransform attribute) (Radial attribute) (SimplexToOrderedTransform attribute) (SoftplusLowerCholeskyTransform attribute) (SoftplusTransform attribute) (Spline attribute) (SplineAutoregressive attribute) (SplineCoupling attribute) (Sylvester attribute) (TransformModule attribute) (UnitLowerCholeskyTransform attribute) DoMessenger (class in pyro.poutine.do_messenger) done (Message attribute) donsker_varadhan_eig() (in module pyro.contrib.oed.eig) DotProduct (class in pyro.contrib.gp.kernels) dtanh_dx() (Sylvester method) DualAveraging (class in pyro.ops.dual_averaging) duration (CoalescentTimesWithRate property) (IndependentHMM property) dynamic_model (EKFState property) DynamicModel (class in pyro.contrib.tracking.dynamic_models)

E

easy_guide() (in module pyro.contrib.easyguide) EasyGuide (class in pyro.contrib.easyguide) edge_mean (SpanningTree property) edges (Trace property) effectful() (in module pyro.poutine.runtime) effective_sample_size() (in module pyro.ops.stats) einsum() (in module pyro.ops.contract) EKFDistribution (class in pyro.contrib.tracking.distributions) EKFState (class in pyro.contrib.tracking.extended_kalman_filter) ELBO (class in pyro.contrib.funsor.infer.elbo) (class in pyro.infer.elbo) elbo() (in module pyro.contrib.minipyro) ELBOModule (class in pyro.infer.elbo) elu() (in module pyro.distributions.transforms) ELUTransform (class in pyro.distributions.transforms) embed() (FactorMuE method) Empirical (class in pyro.distributions) empirical (Marginals property) EmpiricalMarginal (class in pyro.infer.abstract_infer) enable_validation() (in module pyro.contrib.funsor) (in module pyro.infer.util) (in module pyro.poutine.util) (in module pyro.primitives) end_adaptation() (BlockMassMatrix method) end_warmup() (MCMCKernel method) energy_score_empirical() (in module pyro.ops.stats) EnergyDistance (class in pyro.infer.energy_distance) entropy() (Logistic method) enum() (in module pyro.contrib.funsor.handlers) (in module pyro.poutine.handlers) enum_extend() (in module pyro.poutine.util) enumerate (InferDict attribute) enumerate_site() (in module pyro.poutine.enum_messenger) enumerate_support() (BetaBinomial method) (Distribution method) (Empirical method) (MaskedDistribution method) (OneOneMatching method) (OneTwoMatching method) (SpanningTree method) EnumMessenger (class in pyro.poutine.enum_messenger) eps (NeuralAutoregressive attribute) escape() (in module pyro.poutine.handlers) EscapeMessenger (class in pyro.poutine.escape_messenger) eval_crps() (in module pyro.contrib.forecast.evaluate) eval_mae() (in module pyro.contrib.forecast.evaluate) eval_rmse() (in module pyro.contrib.forecast.evaluate) evaluate() (FactorMuE method) (ProfileHMM method) evaluate_loss() (SVI method) event_dim (PyroParam attribute) (TorchDistributionMixin property) event_logsumexp() (Gaussian method) event_pad() (Gaussian method) event_permute() (AffineNormal method) (Gaussian method)

event_shape (Empirical property) (TorchDistributionMixin property) exp_goodness_of_fit() (in module pyro.distributions.testing.gof) expand (InferDict attribute) expand() (AffineBeta method) (AffineNormal method) (AsymmetricLaplace method) (BetaBinomial method) (CoalescentTimesWithRate method) (Delta method) (DirichletMultinomial method) (DiscreteHMM method) (FoldedDistribution method) (GammaGaussianHMM method) (GammaPoisson method) (Gaussian method) (GaussianHMM method) (GaussianMRF method) (GroupedNormalNormal method) (ImproperUniform method) (IndependentHMM method) (InverseGamma method) (LinearHMM method) (LKJ method) (Logistic method) (LogNormalNegativeBinomial method) (MaskedDistribution method) (MaskedMixture method) (MixtureOfDiagNormals method) (MixtureOfDiagNormalsSharedCovariance method) (MultivariateStudentT method) (OrderedLogistic method) (ProjectedNormal method) (SineBivariateVonMises method) (SineSkewed method) (SkewLogistic method) (SoftAsymmetricLaplace method) (SoftLaplace method) (Stable method) (TorchDistribution method) (TorchDistributionMixin method) (TruncatedPolyaGamma method) (Unit method) (VonMises3D method) (ZeroInflatedDistribution method) expand_by() (TorchDistributionMixin method) Exponent (class in pyro.contrib.gp.kernels) Exponential (class in pyro.contrib.gp.kernels) (class in pyro.distributions) ExponentialFamily (class in pyro.distributions) ExponentialLR() (in module pyro.optim.pytorch_optimizers) ExponentialNet (class in pyro.contrib.cevae) ExtendedBetaBinomial (class in pyro.distributions) ExtendedBinomial (class in pyro.distributions)

F

factor() (in module pyro.contrib.funsor) (in module pyro.primitives) FactorMuE (class in pyro.contrib.mue.models) filter() (DiscreteHMM method) (GammaGaussianHMM method) (GaussianHMM method) (MissingDataDiscreteHMM method) filter_states() (EKFDistribution method) finalize() (CompartmentalModel method) FisherSnedecor (class in pyro.distributions) fit() (CEVAE method) fit_generalized_pareto() (in module pyro.ops.stats) fit_mcmc() (CompartmentalModel method) fit_svi() (CompartmentalModel method) (FactorMuE method) (ProfileHMM method) fn (Message attribute) (ReparamMessage attribute) (ReparamResult attribute) FoldedDistribution (class in pyro.distributions) forecast() (DependentMaternGP method) (GenericLGSSM method) (GenericLGSSMWithGPNoiseModel method) (IndependentMaternGP method) (LinearlyCoupledMaternGP method) (TimeSeriesModel method) Forecaster (class in pyro.contrib.forecast.forecaster) ForecastingModel (class in pyro.contrib.forecast.forecaster) format_shapes() (Trace method) forward() (AutoCallable method) (AutoContinuous method) (AutoDelta method) (AutoDiscreteParallel method) (AutoGaussian method) (AutoGuideList method) (AutoNormal method) (AutoRegressiveNN method) (AutoStructured method) (BernoulliNet method) (Binary method) (Brownian method) (ConditionalAutoRegressiveNN method) (ConditionalDenseNN method) (Constant method) (Coregionalize method) (Cosine method) (DenseNN method) (DiagNormalNet method) (DynamicModel method) (EasyGuide method) (ELBOModule method) (Exponent method) (Exponential method) (ExponentialNet method) (Gaussian method) (GPLVM method) (GPModel method) (GPRegression method) (Guide method) (Kernel method) (LaplaceNet method) (Likelihood method) (Linear method) (Matern32 method) (Matern52 method) (MHResampler method) (Model method) (MultiClass method) (Ncp method) (Ncv method) (NormalNet method) (Periodic method) (Poisson method) (Polynomial method) (Predictive method) (Product method) (Profile method) (RationalQuadratic method) (RBF method) (SparseGPRegression method) (StudentTNet method) (Sum method) (VariationalGP method) (VariationalSparseGP method) (VerticalScaling method) (Warping method) (WeighedPredictive method) (WhiteNoise method)

forward_shape() (DiscreteCosineTransform method) (HaarTransform method) (PositivePowerTransform method) (SimplexToOrderedTransform method) frame_num (EKFState property) (Measurement property) full_mass (CompartmentalModel property) full_size (CondIndepStackFrame attribute) FullyConnected (class in pyro.contrib.cevae)

G

Gamma (class in pyro.distributions) GammaGaussianHMM (class in pyro.distributions) GammaPoisson (class in pyro.distributions) gate (ZeroInflatedDistribution property) gate_logits (ZeroInflatedDistribution property) Gaussian (class in pyro.contrib.gp.likelihoods) (class in pyro.ops.gaussian) gaussian_tensordot() (in module pyro.ops.gaussian) GaussianHMM (class in pyro.distributions) GaussianMRF (class in pyro.distributions) GaussianScaleMixture (class in pyro.distributions) gelman_rubin() (in module pyro.ops.stats) generalized_channel_permute() (in module pyro.distributions.transforms) GeneralizedChannelPermute (class in pyro.distributions.transforms) generate() (CompartmentalModel method) generator() (MarkovMessenger method) GenericLGSSM (class in pyro.contrib.timeseries.lgssm) GenericLGSSMWithGPNoiseModel (class in pyro.contrib.timeseries.lgssmgp) geodesic_difference() (DynamicModel method) (Measurement method) Geometric (class in pyro.distributions) get() (CountMeanStats method) (CountMeanVarianceStats method) (CountStats method) (in module pyro.settings) (StackStats method) (StatsOfDict method) (StreamingStats method) get_all_param_names() (ParamStoreDict method) get_base_dist() (AutoContinuous method) (AutoDiagonalNormal method) (AutoMultivariateNormal method) (AutoNormalizingFlow method) get_class() (DistributionNet static method) get_covariance() (WelfordArrowheadCovariance method) (WelfordCovariance method) get_data_directory() (in module pyro.contrib.examples.util) get_data_loader() (in module pyro.contrib.examples.util) get_deltas() (AutoStructured method) get_dependencies() (in module pyro.infer.inspect) get_dist() (DependentMaternGP method) (GenericLGSSM method) (GenericLGSSMWithGPNoiseModel method) (IndependentMaternGP method) (LinearlyCoupledMaternGP method) (TimeSeriesModel method) get_empirical() (SMCFilter method) get_ESS() (LogWeightsMixin method) get_log_normalizer() (LogWeightsMixin method) get_mask() (in module pyro.poutine.runtime) get_min_sample_transition_count() (MHResampler method) get_named_particles() (SVGD method) get_normalized_weights() (LogWeightsMixin method) get_param() (ParamStoreDict method) get_param_store() (in module pyro.contrib.funsor) (in module pyro.contrib.minipyro) (in module pyro.primitives) get_permutation() (ConditionalAutoRegressiveNN method)

get_plates() (in module pyro.poutine.runtime) get_posterior() (AutoContinuous method) (AutoDiagonalNormal method) (AutoHierarchicalNormalMessenger method) (AutoLaplaceApproximation method) (AutoLowRankMultivariateNormal method) (AutoMultivariateNormal method) (AutoNormalizingFlow method) (AutoNormalMessenger method) (AutoRegressiveMessenger method) (GroupedNormalNormal method) (GuideMessenger method) get_quad_rule() (in module pyro.ops.special) get_samples() (MCMC method) (MHResampler method) (Predictive method) get_source_samples() (MHResampler method) get_state() (DualAveraging method) (ParamStoreDict method) (PyroOptim method) get_statistics() (StreamingMCMC method) get_step() (MixedMultiOptimizer method) (MultiOptimizer method) (Newton method) get_stored_samples() (MHResampler method) get_total_transition_count() (MHResampler method) get_trace() (trace method) (TraceHandler method) (TraceMessenger method) get_traces() (GuideMessenger method) get_transform() (AutoContinuous method) (AutoDiagonalNormal method) (AutoMultivariateNormal method) (AutoNormalizingFlow method) get_vectorized_trace() (Predictive method) given_states() (MissingDataDiscreteHMM method) GLOBAL (DimType attribute) global_frame (DimStack property) global_model() (CompartmentalModel method) GlobalNamedMessenger (class in pyro.contrib.funsor.handlers.named_messenger) GPLVM (class in pyro.contrib.gp.models.gplvm) GPModel (class in pyro.contrib.gp.models.model) GPRegression (class in pyro.contrib.gp.models.gpr) Group (class in pyro.contrib.easyguide.easyguide) group() (EasyGuide method) GroupedNormalNormal (class in pyro.distributions) Guide (class in pyro.contrib.cevae) guide (Group property) guide() (EasyGuide method) (GPLVM method) (GPModel method) (GPRegression method) (SparseGPRegression method) (VariationalGP method) (VariationalSparseGP method) guide_log_prob (WeighedPredictiveResults attribute) GuideMessenger (class in pyro.poutine.guide) Gumbel (class in pyro.distributions) GumbelSoftmaxReparam (class in pyro.infer.reparam.softmax)

H

haar_transform() (in module pyro.ops.tensor_utils) HaarReparam (class in pyro.infer.reparam.haar) HaarTransform (class in pyro.distributions.transforms) HalfCauchy (class in pyro.distributions) HalfNormal (class in pyro.distributions) has_enumerate_support (BetaBinomial attribute) (Distribution attribute) (Empirical attribute) (MaskedDistribution property) (OneOneMatching attribute) (OneTwoMatching attribute) (SpanningTree attribute) has_rsample (AsymmetricLaplace attribute) (Delta attribute) (Distribution attribute) (GaussianHMM attribute) (GaussianScaleMixture attribute) (IndependentHMM property) (InverseGamma attribute) (LinearHMM attribute) (Logistic attribute) (MaskedDistribution property) (MaskedMixture property) (MixtureOfDiagNormals attribute) (MixtureOfDiagNormalsSharedCovariance attribute) (MultivariateStudentT attribute) (ProjectedNormal attribute) (Rejector attribute) (SkewLogistic attribute) (SoftAsymmetricLaplace attribute) (SoftLaplace attribute) (Stable attribute) (TruncatedPolyaGamma attribute)

has_rsample_() (Distribution method) HeterogeneousRegionalSIRModel (class in pyro.contrib.epidemiology.models) HeterogeneousSIRModel (class in pyro.contrib.epidemiology.models) heuristic() (CompartmentalModel method) HiddenLayer (class in pyro.contrib.bnn.hidden_layer) high (AffineBeta property) HMC (class in pyro.infer.mcmc) HMCForecaster (class in pyro.contrib.forecast.forecaster) HorovodOptimizer (class in pyro.optim.horovod) Householder (class in pyro.distributions.transforms) householder() (in module pyro.distributions.transforms) hpdi() (in module pyro.ops.stats)

I

icdf() (Logistic method) (SkewLogistic method) (SoftLaplace method) idct() (in module pyro.ops.tensor_utils) identify_dense_edges() (in module pyro.poutine.trace_messenger) Importance (class in pyro.infer.importance) ImproperUniform (class in pyro.distributions.improper_uniform) IMQSteinKernel (class in pyro.infer.svgd) imresize() (in module pyro.contrib.examples.multi_mnist) Independent (class in pyro.distributions) independent() (TorchDistributionMixin method) IndependentHMM (class in pyro.distributions) IndependentMaternGP (class in pyro.contrib.timeseries.gp) IndepMessenger (class in pyro.poutine.indep_messenger) Index (class in pyro.ops.indexing) index() (in module pyro.ops.indexing) indices (IndepMessenger property) infection_dist() (in module pyro.contrib.epidemiology.distributions) infer (Message attribute) infer_config() (in module pyro.poutine.handlers) infer_discrete() (in module pyro.contrib.funsor.infer.discrete) (in module pyro.infer.discrete) infer_shapes() (AffineBeta static method) (DirichletMultinomial static method) (MultivariateStudentT static method) (ProjectedNormal static method) (SineBivariateVonMises class method) (TorchDistributionMixin class method) InferConfigMessenger (class in pyro.poutine.infer_config_messenger) InferDict (class in pyro.poutine.runtime) information_criterion() (TracePosterior method) init() (EasyGuide method) (SMCFilter method) init_to_feasible() (in module pyro.infer.autoguide.initialization) init_to_generated() (in module pyro.infer.autoguide.initialization) init_to_mean() (in module pyro.infer.autoguide.initialization)

init_to_median() (in module pyro.infer.autoguide.initialization) init_to_sample() (in module pyro.infer.autoguide.initialization) init_to_uniform() (in module pyro.infer.autoguide.initialization) init_to_value() (in module pyro.infer.autoguide.initialization) init_value (PyroParam attribute) initial_params (HMC property) (MCMCKernel property) (RandomWalkKernel property) initialize() (CompartmentalModel method) initialize_model() (in module pyro.infer.mcmc.util) InitMessenger (class in pyro.infer.autoguide.initialization) innovation() (EKFState method) inv (ConditionalTransformModule property) inv_permutation (Permute property) inverse_haar_transform() (in module pyro.ops.tensor_utils) inverse_mass_matrix (BlockMassMatrix property) (HMC property) inverse_shape() (DiscreteCosineTransform method) (HaarTransform method) (PositivePowerTransform method) (SimplexToOrderedTransform method) InverseGamma (class in pyro.distributions) is_auxiliary (InferDict attribute) is_observed (InferDict attribute) (Message attribute) (ReparamMessage attribute) (ReparamResult attribute) is_scheduler() (in module pyro.optim.optim) is_validation_enabled() (in module pyro.infer.util) (in module pyro.poutine.util) Isotropy (class in pyro.contrib.gp.kernels) ite() (CEVAE method) items() (ParamStoreDict method) iter_sample() (GPRegression method) iter_stochastic_nodes() (Trace method) iterated() (in module pyro.distributions.transforms)

J

jacobian() (DifferentiableDynamicModel method) (DifferentiableMeasurement method) (Ncp method) (Ncv method) (PositionMeasurement method) Jit_ELBO (class in pyro.contrib.funsor.infer.elbo) JitTrace_ELBO (class in pyro.contrib.funsor.infer.trace_elbo) (class in pyro.contrib.minipyro) (class in pyro.infer.trace_elbo)

JitTraceEnum_ELBO (class in pyro.contrib.funsor.infer.traceenum_elbo) (class in pyro.infer.traceenum_elbo) JitTraceGraph_ELBO (class in pyro.infer.tracegraph_elbo) JitTraceMarkovEnum_ELBO (class in pyro.contrib.funsor.infer.traceenum_elbo) JitTraceMeanField_ELBO (class in pyro.infer.trace_mean_field_elbo) JitTraceTMC_ELBO (class in pyro.contrib.funsor.infer.tracetmc_elbo)

K

Kernel (class in pyro.contrib.gp.kernels) keys() (ParamStoreDict method)

kinetic_grad() (BlockMassMatrix method) Kumaraswamy (class in pyro.distributions) kwargs (Message attribute)

L

LambdaLR() (in module pyro.optim.pytorch_optimizers) Laplace (class in pyro.distributions) laplace_approximation() (AutoLaplaceApproximation method) laplace_eig() (in module pyro.contrib.oed.eig) LaplaceNet (class in pyro.contrib.cevae) LatentStableReparam (class in pyro.infer.reparam.stable) leaky_relu() (in module pyro.distributions.transforms) LeakyReLUTransform (class in pyro.distributions.transforms) left_condition() (AffineNormal method) (Gaussian method) left_scale (AsymmetricLaplace property) (SoftAsymmetricLaplace property) lfire_eig() (in module pyro.contrib.oed.eig) lift() (in module pyro.poutine.handlers) LiftMessenger (class in pyro.poutine.lift_messenger) Likelihood (class in pyro.contrib.gp.likelihoods) Linear (class in pyro.contrib.gp.kernels) LinearHMM (class in pyro.distributions) LinearHMMReparam (class in pyro.infer.reparam.hmm) LinearLR() (in module pyro.optim.pytorch_optimizers) LinearlyCoupledMaternGP (class in pyro.contrib.timeseries.gp) List (class in pyro.contrib.autoname.named) LKJ (class in pyro.distributions) LKJCholesky (class in pyro.distributions) LKJCorrCholesky (class in pyro.distributions) load() (in module pyro.contrib.examples.multi_mnist) (ParamStoreDict method) (PyroOptim method) load_bart_od() (in module pyro.contrib.examples.bart) load_fake_od() (in module pyro.contrib.examples.bart) load_mnist() (in module pyro.contrib.examples.multi_mnist) load_nextstrain_counts() (in module pyro.contrib.examples.nextstrain) loc (AffineBeta property) LOCAL (DimType attribute) local_frame (DimStack property) LocScaleReparam (class in pyro.infer.reparam.loc_scale) log_abs_det_jacobian() (AffineAutoregressive method) (AffineCoupling method) (BatchNorm method) (BlockAutoregressive method) (CholeskyTransform method) (CorrMatrixCholeskyTransform method) (DiscreteCosineTransform method) (ELUTransform method) (HaarTransform method) (LeakyReLUTransform method) (LowerCholeskyAffine method) (NeuralAutoregressive method) (OrderedTransform method) (Permute method) (Polynomial method) (PositivePowerTransform method) (SimplexToOrderedTransform method) (SoftplusTransform method) (SplineAutoregressive method) (SplineCoupling method) (Sylvester method) log_beta() (in module pyro.ops.special) log_binomial() (in module pyro.ops.special) log_density() (Gaussian method) log_I1() (in module pyro.ops.special) log_kernel_and_grad() (IMQSteinKernel method) (RBFSteinKernel method) (SteinKernel method) log_likelihood_of_update() (EKFState method) log_partition_function (OneOneMatching property) (OneTwoMatching property) (SpanningTree property) log_prob (Message attribute) log_prob() (AsymmetricLaplace method) (BetaBinomial method) (CoalescentTimes method) (CoalescentTimesWithRate method) (Delta method) (DependentMaternGP method) (DirichletMultinomial method) (DiscreteHMM method) (Distribution method) (EKFDistribution method) (Empirical method) (ExtendedBetaBinomial method) (ExtendedBinomial method) (FoldedDistribution method) (GammaGaussianHMM method) (GammaPoisson method) (GaussianHMM method) (GaussianMRF method) (GaussianScaleMixture method) (GenericLGSSM method) (GenericLGSSMWithGPNoiseModel method) (GroupedNormalNormal method) (ImproperUniform method) (IndependentHMM method) (IndependentMaternGP method) (LinearHMM method) (LinearlyCoupledMaternGP method) (Logistic method) (LogNormalNegativeBinomial method) (MaskedDistribution method) (MaskedMixture method) (MissingDataDiscreteHMM method) (MixtureOfDiagNormals method) (MixtureOfDiagNormalsSharedCovariance method) (MultivariateStudentT method) (NanMaskedMultivariateNormal method) (NanMaskedNormal method) (OneOneMatching method) (OneTwoMatching method) (ProjectedNormal method) (Rejector method) (RelaxedBernoulliStraightThrough method) (RelaxedOneHotCategoricalStraightThrough method) (SineBivariateVonMises method) (SineSkewed method) (SkewLogistic method) (SoftAsymmetricLaplace method) (SoftLaplace method) (SpanningTree method) (Stable method) (TimeSeriesModel method) (TruncatedPolyaGamma method) (Unit method) (VonMises3D method) (ZeroInflatedDistribution method)

log_prob_sum (Message attribute) log_prob_sum() (Trace method) log_weights (Empirical property) (Importance attribute) (LogWeightsMixin attribute) (WeighedPredictiveResults attribute) logging() (HMC method) (MCMCKernel method) (RandomWalkKernel method) Logistic (class in pyro.distributions) LogisticNormal (class in pyro.distributions) logits (ZeroInflatedNegativeBinomial property) LogNormal (class in pyro.distributions) LogNormalNegativeBinomial (class in pyro.distributions) LogWeightsMixin (class in pyro.infer.importance) loss() (ELBO method) (EnergyDistance method) (RenyiELBO method) (ReweightedWakeSleep method) (Trace_ELBO method) (TraceCausalEffect_ELBO method) (TraceEnum_ELBO method) (TraceGraph_ELBO method) (TraceMeanField_ELBO method) (TraceTailAdaptive_ELBO method) (TraceTMC_ELBO method) loss_and_grads() (ELBO method) (JitTrace_ELBO method) (JitTraceEnum_ELBO method) (JitTraceGraph_ELBO method) (JitTraceMeanField_ELBO method) (RenyiELBO method) (ReweightedWakeSleep method) (Trace_ELBO method) (TraceEnum_ELBO method) (TraceGraph_ELBO method) (TraceTMC_ELBO method) loss_and_surrogate_loss() (JitTrace_ELBO method) low (AffineBeta property) LowerCholeskyAffine (class in pyro.distributions.transforms) LowRankMultivariateNormal (class in pyro.distributions) LRScheduler() (in module pyro.optim.pytorch_optimizers) LSH (class in pyro.contrib.tracking.hashing)

M

make_dist() (BernoulliNet static method) (ExponentialNet static method) (LaplaceNet static method) (NormalNet static method) (StudentTNet static method) map_estimate() (EasyGuide method) (Group method) map_states() (MissingDataDiscreteHMM method) marginal() (TracePosterior method) (TracePredictive method) marginal_eig() (in module pyro.contrib.oed.eig) MarginalAssignment (class in pyro.contrib.tracking.assignment) MarginalAssignmentPersistent (class in pyro.contrib.tracking.assignment) MarginalAssignmentSparse (class in pyro.contrib.tracking.assignment) marginalize() (AffineNormal method) (Gaussian method) Marginals (class in pyro.infer.abstract_infer) markov() (in module pyro.contrib.funsor) (in module pyro.contrib.funsor.handlers) (in module pyro.poutine.handlers) MarkovMessenger (class in pyro.contrib.funsor.handlers.named_messenger) (class in pyro.poutine.markov_messenger) mask (Message attribute) mask() (in module pyro.poutine.handlers) (TorchDistributionMixin method) MaskedDistribution (class in pyro.distributions) MaskedMixture (class in pyro.distributions) MaskMessenger (class in pyro.poutine.mask_messenger) mass_matrix_adapter (HMC property) mass_matrix_size (BlockMassMatrix property) match() (ParamStoreDict method) Matern32 (class in pyro.contrib.gp.kernels) Matern52 (class in pyro.contrib.gp.kernels) MaternKernel (class in pyro.ops.ssm_gp) matmul() (in module pyro.ops.tensor_utils) matrix_and_gaussian_to_gaussian() (in module pyro.ops.gaussian) matrix_and_mvn_to_gaussian() (in module pyro.ops.gaussian) matrix_exponential() (in module pyro.distributions.transforms) MatrixExponential (class in pyro.distributions.transforms) matvecmul() (in module pyro.ops.tensor_utils) MAX_DIM (DimStack attribute) max_sample_iter (SineBivariateVonMises attribute) mc_extend() (in module pyro.poutine.util) MCMC (class in pyro.infer.mcmc.api) MCMCKernel (class in pyro.infer.mcmc.mcmc_kernel) mean (AffineBeta property) (AsymmetricLaplace property) (BetaBinomial property) (Delta property) (DirichletMultinomial property) (EKFState property) (Empirical property) (GammaPoisson property) (LKJ property) (Logistic property) (LogNormalNegativeBinomial property) (MaskedDistribution property) (MaskedMixture property) (Measurement property) (MultivariateStudentT property) (ProjectedNormal property) (SineBivariateVonMises property) (SoftAsymmetricLaplace property) (SoftLaplace property) (Stable property) (ZeroInflatedDistribution property) mean2pv() (DynamicModel method) (Ncp method) (Ncv method) mean_pv (EKFState property) Measurement (class in pyro.contrib.tracking.measurements) median() (AutoContinuous method) (AutoDelta method) (AutoGaussian method) (AutoGuide method) (AutoGuideList method) (AutoHierarchicalNormalMessenger method) (AutoNormal method) (AutoNormalMessenger method) (AutoStructured method) merge() (CountMeanStats method) (CountMeanVarianceStats method) (CountStats method) (StackStats method) (StatsOfDict method) (StreamingStats method) merge_points() (in module pyro.contrib.tracking.hashing) Message (class in pyro.poutine.runtime) Messenger (class in pyro.contrib.minipyro) (class in pyro.poutine.messenger) mg2k() (in module pyro.contrib.mue.statearrangers) MHResampler (class in pyro.infer.predictive) MinimalReparam (class in pyro.infer.reparam.strategies) mirrors (MNIST attribute) MissingDataDiscreteHMM (class in pyro.contrib.mue.missingdatahmm) MixedMultiOptimizer (class in pyro.optim.multi) MixtureOfDiagNormals (class in pyro.distributions) MixtureOfDiagNormalsSharedCovariance (class in pyro.distributions) MixtureSameFamily (class in pyro.distributions) mk_dataset() (in module pyro.contrib.examples.multi_mnist) MNIST (class in pyro.contrib.examples.util) mode (Parameterized property), [1] (ProjectedNormal property) (SpanningTree property) mode() (OneOneMatching method) (OneTwoMatching method) model (AutoGuide property) Model (class in pyro.contrib.cevae) model (EasyGuide property) (GuideMessenger property) model() (ForecastingModel method) (GPLVM method) (GPModel method) (GPRegression method) (SparseGPRegression method) (VariationalGP method) (VariationalSparseGP method) model_log_prob (WeighedPredictiveResults attribute) module pyro.contrib.autoname pyro.contrib.autoname.named pyro.contrib.autoname.scoping pyro.contrib.bnn pyro.contrib.bnn.hidden_layer pyro.contrib.cevae pyro.contrib.easyguide pyro.contrib.epidemiology pyro.contrib.epidemiology.compartmental pyro.contrib.epidemiology.distributions pyro.contrib.epidemiology.models pyro.contrib.examples.bart pyro.contrib.examples.multi_mnist pyro.contrib.examples.nextstrain pyro.contrib.examples.util pyro.contrib.forecast pyro.contrib.forecast.evaluate pyro.contrib.forecast.forecaster pyro.contrib.funsor pyro.contrib.funsor.handlers pyro.contrib.funsor.handlers.named_messenger pyro.contrib.funsor.handlers.primitives pyro.contrib.funsor.handlers.runtime pyro.contrib.funsor.infer.discrete pyro.contrib.funsor.infer.elbo pyro.contrib.funsor.infer.trace_elbo pyro.contrib.funsor.infer.traceenum_elbo pyro.contrib.funsor.infer.tracetmc_elbo pyro.contrib.gp pyro.contrib.gp.kernels pyro.contrib.gp.likelihoods pyro.contrib.gp.models.gplvm pyro.contrib.gp.models.gpr pyro.contrib.gp.models.model pyro.contrib.gp.models.sgpr pyro.contrib.gp.models.vgp pyro.contrib.gp.models.vsgp pyro.contrib.gp.parameterized pyro.contrib.gp.util pyro.contrib.minipyro pyro.contrib.mue pyro.contrib.mue.dataloaders pyro.contrib.mue.missingdatahmm pyro.contrib.mue.models pyro.contrib.mue.statearrangers pyro.contrib.oed pyro.contrib.oed.eig pyro.contrib.oed.glmm pyro.contrib.randomvariable pyro.contrib.timeseries pyro.contrib.timeseries.base pyro.contrib.timeseries.gp pyro.contrib.timeseries.lgssm pyro.contrib.timeseries.lgssmgp pyro.contrib.tracking pyro.contrib.tracking.assignment pyro.contrib.tracking.distributions pyro.contrib.tracking.dynamic_models pyro.contrib.tracking.extended_kalman_filter pyro.contrib.tracking.hashing pyro.contrib.tracking.measurements pyro.contrib.zuko pyro.distributions.constraints pyro.distributions.testing.gof pyro.distributions.torch pyro.infer.abstract_infer pyro.infer.autoguide pyro.infer.autoguide.initialization pyro.infer.discrete pyro.infer.elbo pyro.infer.energy_distance pyro.infer.importance pyro.infer.inspect pyro.infer.predictive pyro.infer.renyi_elbo pyro.infer.reparam pyro.infer.reparam.conjugate pyro.infer.reparam.discrete_cosine pyro.infer.reparam.haar pyro.infer.reparam.hmm pyro.infer.reparam.loc_scale pyro.infer.reparam.neutra pyro.infer.reparam.projected_normal pyro.infer.reparam.reparam pyro.infer.reparam.softmax pyro.infer.reparam.split pyro.infer.reparam.stable pyro.infer.reparam.strategies pyro.infer.reparam.structured pyro.infer.reparam.studentt pyro.infer.reparam.transform pyro.infer.reparam.unit_jacobian pyro.infer.resampler pyro.infer.rws pyro.infer.smcfilter pyro.infer.svgd pyro.infer.svi pyro.infer.trace_elbo pyro.infer.trace_mean_field_elbo pyro.infer.trace_tail_adaptive_elbo pyro.infer.traceenum_elbo pyro.infer.tracegraph_elbo pyro.infer.tracetmc_elbo pyro.nn.module pyro.ops.dual_averaging pyro.ops.einsum pyro.ops.gaussian pyro.ops.indexing pyro.ops.integrator pyro.ops.newton pyro.ops.special pyro.ops.ssm_gp pyro.ops.stats pyro.ops.streaming pyro.ops.tensor_utils pyro.ops.welford pyro.optim.adagrad_rmsprop pyro.optim.clipped_adam pyro.optim.horovod pyro.optim.lr_scheduler pyro.optim.multi pyro.optim.optim pyro.optim.pytorch_optimizers pyro.params.param_store pyro.poutine.block_messenger pyro.poutine.broadcast_messenger pyro.poutine.collapse_messenger pyro.poutine.condition_messenger pyro.poutine.do_messenger pyro.poutine.enum_messenger pyro.poutine.escape_messenger pyro.poutine.guide pyro.poutine.handlers pyro.poutine.indep_messenger pyro.poutine.infer_config_messenger pyro.poutine.lift_messenger pyro.poutine.markov_messenger pyro.poutine.mask_messenger pyro.poutine.messenger pyro.poutine.plate_messenger pyro.poutine.reentrant_messenger pyro.poutine.reparam_messenger pyro.poutine.replay_messenger pyro.poutine.runtime pyro.poutine.scale_messenger pyro.poutine.seed_messenger pyro.poutine.subsample_messenger pyro.poutine.substitute_messenger pyro.poutine.trace_messenger pyro.poutine.uncondition_messenger pyro.poutine.util pyro.primitives pyro.settings

module() (in module pyro.contrib.funsor) (in module pyro.primitives) module_from_param_with_module_name() (in module pyro.params.param_store) MultiClass (class in pyro.contrib.gp.likelihoods) Multinomial (class in pyro.distributions) multinomial_goodness_of_fit() (in module pyro.distributions.testing.gof) MultiOptimizer (class in pyro.optim.multi) MultiplicativeLR() (in module pyro.optim.pytorch_optimizers) MultiStepLR() (in module pyro.optim.pytorch_optimizers) MultivariateNormal (class in pyro.distributions) MultivariateStudentT (class in pyro.distributions) mvn_to_gaussian() (in module pyro.ops.gaussian)

N

NAdam() (in module pyro.optim.pytorch_optimizers) name (CondIndepStackFrame attribute) (Message attribute) (ReparamMessage attribute) name_count() (in module pyro.contrib.autoname) (in module pyro.contrib.autoname.scoping) NameCountMessenger (class in pyro.contrib.autoname.scoping) named() (in module pyro.contrib.funsor.handlers) named_parameters() (ParamStoreDict method) named_pyro_params() (PyroModule method) NamedMessenger (class in pyro.contrib.funsor.handlers.named_messenger) names_from_batch_shape() (DimStack method) NanMaskedMultivariateNormal (class in pyro.distributions) NanMaskedNormal (class in pyro.distributions) Ncp (class in pyro.contrib.tracking.dynamic_models) NcpContinuous (class in pyro.contrib.tracking.dynamic_models) NcpDiscrete (class in pyro.contrib.tracking.dynamic_models) Ncv (class in pyro.contrib.tracking.dynamic_models) NcvContinuous (class in pyro.contrib.tracking.dynamic_models) NcvDiscrete (class in pyro.contrib.tracking.dynamic_models) nearby() (LSH method) NegativeBinomial (class in pyro.distributions) neural_autoregressive() (in module pyro.distributions.transforms)

NeuralAutoregressive (class in pyro.distributions.transforms) NeuTraReparam (class in pyro.infer.reparam.neutra) Newton (class in pyro.optim.multi) newton_step() (in module pyro.ops.newton) newton_step_1d() (in module pyro.ops.newton) newton_step_2d() (in module pyro.ops.newton) newton_step_3d() (in module pyro.ops.newton) next_context() (IndepMessenger method) next_fast_len() (in module pyro.ops.tensor_utils) nmc_eig() (in module pyro.contrib.oed.eig) NonlocalExit nonreparam_stochastic_nodes (Trace property) norm_const (SineBivariateVonMises property) Normal (class in pyro.distributions) Normalize (class in pyro.distributions.transforms) normalize_param_name() (in module pyro.params.param_store) NormalNet (class in pyro.contrib.cevae) num_gamma_variates (TruncatedPolyaGamma attribute) num_log_prob_terms (TruncatedPolyaGamma attribute) num_process_noise_parameters (DynamicModel property) num_samples (InferDict attribute) num_steps (HMC property) NUTS (class in pyro.infer.mcmc)

O

Object (class in pyro.contrib.autoname.named) obs (InferDict attribute) (Message attribute) observation_nodes (Trace property) OMTMultivariateNormal (class in pyro.distributions) OneCycleLR() (in module pyro.optim.pytorch_optimizers) OneHotCategorical (class in pyro.distributions)

OneHotCategoricalStraightThrough (class in pyro.distributions) OneOneMatching (class in pyro.distributions) OneTwoMatching (class in pyro.distributions) OrderedLogistic (class in pyro.distributions) OrderedTransform (class in pyro.distributions.transforms) OverdispersedSEIRModel (class in pyro.contrib.epidemiology.models) OverdispersedSIRModel (class in pyro.contrib.epidemiology.models)

P

pack_tensors() (Trace method) packed (Message attribute) param() (in module pyro.contrib.funsor) (in module pyro.contrib.minipyro) (in module pyro.primitives) param_() (Object method) param_name() (ParamStoreDict method) param_nodes (Trace property) param_with_module_name() (in module pyro.params.param_store) Parameterized (class in pyro.contrib.gp) (class in pyro.contrib.gp.parameterized) params (StateDict attribute) ParamStoreDict (class in pyro.params.param_store) Pareto (class in pyro.distributions) Periodic (class in pyro.contrib.gp.kernels) periodic_cumsum() (in module pyro.ops.tensor_utils) periodic_features() (in module pyro.ops.tensor_utils) periodic_repeat() (in module pyro.ops.tensor_utils) Permute (class in pyro.distributions.transforms) permute() (in module pyro.distributions.transforms) pi() (in module pyro.ops.stats) Planar (class in pyro.distributions.transforms) planar() (in module pyro.distributions.transforms) plate (class in pyro.primitives) plate() (EasyGuide method) (in module pyro.contrib.funsor.handlers) (in module pyro.contrib.minipyro) plate_stack() (in module pyro.primitives) PlateMessenger (class in pyro.contrib.minipyro) (class in pyro.poutine.plate_messenger) Poisson (class in pyro.contrib.gp.likelihoods) (class in pyro.distributions) Polynomial (class in pyro.contrib.gp.kernels) (class in pyro.distributions.transforms) polynomial() (in module pyro.distributions.transforms) PolynomialLR() (in module pyro.optim.pytorch_optimizers) pop_global() (DimStack method) pop_iter() (DimStack method) pop_local() (DimStack method) PositionMeasurement (class in pyro.contrib.tracking.measurements) PositivePowerTransform (class in pyro.distributions.transforms) posterior_eig() (in module pyro.contrib.oed.eig) postprocess_message() (Messenger method) (trace method) potential_grad() (in module pyro.ops.integrator) precision_matrix (MultivariateStudentT property) precision_to_scale_tril() (in module pyro.ops.tensor_utils) predecessors() (Trace method) predict() (CompartmentalModel method) (EKFState method) (ForecastingModel method) Predictive (class in pyro.infer.predictive) prefix_condition() (GaussianHMM method) print_and_log() (in module pyro.contrib.examples.util) prior (InferDict attribute) (PyroSample attribute) probs (ZeroInflatedNegativeBinomial property) process_covariance() (MaternKernel method) process_message() (block method) (Messenger method) (PlateMessenger method) (replay method) process_noise_cov() (DynamicModel method) (Ncp method) (NcpContinuous method) (NcpDiscrete method) (Ncv method) (NcvContinuous method) (NcvDiscrete method) process_noise_dist() (DynamicModel method) Product (class in pyro.contrib.gp.kernels) Profile (class in pyro.contrib.mue.statearrangers) ProfileHMM (class in pyro.contrib.mue.models) ProjectedNormal (class in pyro.distributions) ProjectedNormalReparam (class in pyro.infer.reparam.projected_normal) prune_subsample_sites() (in module pyro.poutine.util) psis_diagnostic() (in module pyro.infer.importance) push_global() (DimStack method) push_iter() (DimStack method) push_local() (DimStack method) pyro.contrib.autoname module pyro.contrib.autoname.named module pyro.contrib.autoname.scoping module pyro.contrib.bnn module pyro.contrib.bnn.hidden_layer module pyro.contrib.cevae module pyro.contrib.easyguide module pyro.contrib.epidemiology module pyro.contrib.epidemiology.compartmental module pyro.contrib.epidemiology.distributions module pyro.contrib.epidemiology.models module pyro.contrib.examples.bart module pyro.contrib.examples.multi_mnist module pyro.contrib.examples.nextstrain module pyro.contrib.examples.util module pyro.contrib.forecast module pyro.contrib.forecast.evaluate module pyro.contrib.forecast.forecaster module pyro.contrib.funsor module pyro.contrib.funsor.handlers module pyro.contrib.funsor.handlers.named_messenger module pyro.contrib.funsor.handlers.primitives module pyro.contrib.funsor.handlers.runtime module pyro.contrib.funsor.infer.discrete module pyro.contrib.funsor.infer.elbo module pyro.contrib.funsor.infer.trace_elbo module pyro.contrib.funsor.infer.traceenum_elbo module pyro.contrib.funsor.infer.tracetmc_elbo module pyro.contrib.gp module pyro.contrib.gp.kernels module pyro.contrib.gp.likelihoods module pyro.contrib.gp.models.gplvm module pyro.contrib.gp.models.gpr module pyro.contrib.gp.models.model module pyro.contrib.gp.models.sgpr module pyro.contrib.gp.models.vgp module pyro.contrib.gp.models.vsgp module pyro.contrib.gp.parameterized module pyro.contrib.gp.util module pyro.contrib.minipyro module pyro.contrib.mue module pyro.contrib.mue.dataloaders module pyro.contrib.mue.missingdatahmm module pyro.contrib.mue.models module pyro.contrib.mue.statearrangers module pyro.contrib.oed module pyro.contrib.oed.eig module pyro.contrib.oed.glmm module pyro.contrib.randomvariable module pyro.contrib.timeseries module pyro.contrib.timeseries.base module pyro.contrib.timeseries.gp module pyro.contrib.timeseries.lgssm module pyro.contrib.timeseries.lgssmgp module pyro.contrib.tracking module pyro.contrib.tracking.assignment module pyro.contrib.tracking.distributions module pyro.contrib.tracking.dynamic_models module pyro.contrib.tracking.extended_kalman_filter module

pyro.contrib.tracking.hashing module pyro.contrib.tracking.measurements module pyro.contrib.zuko module pyro.distributions.constraints module pyro.distributions.testing.gof module pyro.distributions.torch module pyro.infer.abstract_infer module pyro.infer.autoguide module pyro.infer.autoguide.initialization module pyro.infer.discrete module pyro.infer.elbo module pyro.infer.energy_distance module pyro.infer.importance module pyro.infer.inspect module pyro.infer.predictive module pyro.infer.renyi_elbo module pyro.infer.reparam module pyro.infer.reparam.conjugate module pyro.infer.reparam.discrete_cosine module pyro.infer.reparam.haar module pyro.infer.reparam.hmm module pyro.infer.reparam.loc_scale module pyro.infer.reparam.neutra module pyro.infer.reparam.projected_normal module pyro.infer.reparam.reparam module pyro.infer.reparam.softmax module pyro.infer.reparam.split module pyro.infer.reparam.stable module pyro.infer.reparam.strategies module pyro.infer.reparam.structured module pyro.infer.reparam.studentt module pyro.infer.reparam.transform module pyro.infer.reparam.unit_jacobian module pyro.infer.resampler module pyro.infer.rws module pyro.infer.smcfilter module pyro.infer.svgd module pyro.infer.svi module pyro.infer.trace_elbo module pyro.infer.trace_mean_field_elbo module pyro.infer.trace_tail_adaptive_elbo module pyro.infer.traceenum_elbo module pyro.infer.tracegraph_elbo module pyro.infer.tracetmc_elbo module pyro.nn.module module pyro.ops.dual_averaging module pyro.ops.einsum module pyro.ops.gaussian module pyro.ops.indexing module pyro.ops.integrator module pyro.ops.newton module pyro.ops.special module pyro.ops.ssm_gp module pyro.ops.stats module pyro.ops.streaming module pyro.ops.tensor_utils module pyro.ops.welford module pyro.optim.adagrad_rmsprop module pyro.optim.clipped_adam module pyro.optim.horovod module pyro.optim.lr_scheduler module pyro.optim.multi module pyro.optim.optim module pyro.optim.pytorch_optimizers module pyro.params.param_store module pyro.poutine.block_messenger module pyro.poutine.broadcast_messenger module pyro.poutine.collapse_messenger module pyro.poutine.condition_messenger module pyro.poutine.do_messenger module pyro.poutine.enum_messenger module pyro.poutine.escape_messenger module pyro.poutine.guide module pyro.poutine.handlers module pyro.poutine.indep_messenger module pyro.poutine.infer_config_messenger module pyro.poutine.lift_messenger module pyro.poutine.markov_messenger module pyro.poutine.mask_messenger module pyro.poutine.messenger module pyro.poutine.plate_messenger module pyro.poutine.reentrant_messenger module pyro.poutine.reparam_messenger module pyro.poutine.replay_messenger module pyro.poutine.runtime module pyro.poutine.scale_messenger module pyro.poutine.seed_messenger module pyro.poutine.subsample_messenger module pyro.poutine.substitute_messenger module pyro.poutine.trace_messenger module pyro.poutine.uncondition_messenger module pyro.poutine.util module pyro.primitives module pyro.settings module pyro_method() (in module pyro.nn.module) PyroLRScheduler (class in pyro.optim.lr_scheduler) PyroModule (class in pyro.nn.module) PyroModuleList (class in pyro.nn.module) PyroMultiOptimizer (class in pyro.optim.multi) PyroOptim (class in pyro.optim.optim) PyroParam (class in pyro.nn.module) PyroSample (class in pyro.nn.module)

Q

Q() (Sylvester method) quantile() (in module pyro.ops.stats) quantiles() (AutoContinuous method) (AutoGuideList method) (AutoNormal method)

queue() (in module pyro.poutine.handlers)

R

R() (Sylvester method) RAdam() (in module pyro.optim.pytorch_optimizers) Radial (class in pyro.distributions.transforms) radial() (in module pyro.distributions.transforms) random_module() (in module pyro.contrib.funsor) (in module pyro.primitives) RandomVariable (class in pyro.contrib.randomvariable.random_variable) RandomWalkKernel (class in pyro.infer.mcmc) rate (GammaPoisson property) (InverseGamma property) (ZeroInflatedPoisson property) RationalQuadratic (class in pyro.contrib.gp.kernels) RBF (class in pyro.contrib.gp.kernels) RBFSteinKernel (class in pyro.infer.svgd) ReduceLROnPlateau() (in module pyro.optim.pytorch_optimizers) ReentrantMessenger (class in pyro.poutine.reentrant_messenger) region_plate (CompartmentalModel property) RegionalSIRModel (class in pyro.contrib.epidemiology.models) register() (in module pyro.settings) (Messenger class method) register_exception_handler() (in module pyro.ops.integrator) Rejector (class in pyro.distributions) RelaxedBernoulli (class in pyro.distributions) RelaxedBernoulliStraightThrough (class in pyro.distributions) RelaxedOneHotCategorical (class in pyro.distributions) RelaxedOneHotCategoricalStraightThrough (class in pyro.distributions) remove() (LSH method) remove_node() (Trace method) render_model() (in module pyro.infer.inspect) RenyiELBO (class in pyro.infer.renyi_elbo) Reparam (class in pyro.infer.reparam.reparam) reparam() (in module pyro.poutine.handlers) (NeuTraReparam method) (StructuredReparam method) reparameterized_nodes (Trace property) ReparamHandler (class in pyro.poutine.reparam_messenger) ReparamMessage (class in pyro.infer.reparam.reparam) ReparamMessenger (class in pyro.poutine.reparam_messenger) ReparamResult (class in pyro.infer.reparam.reparam) repeated_matmul() (in module pyro.ops.tensor_utils) replace_param() (ParamStoreDict method) replay (class in pyro.contrib.minipyro) replay() (in module pyro.contrib.funsor.handlers) (in module pyro.poutine.handlers) ReplayMessenger (class in pyro.poutine.replay_messenger) resample() (in module pyro.ops.stats) Resampler (class in pyro.infer.resampler) reset() (DualAveraging method) (WelfordArrowheadCovariance method) (WelfordCovariance method)

reset_parameters() (Householder method) (MatrixExponential method) (Planar method) (Polynomial method) (Radial method) reset_parameters2() (Sylvester method) reset_stack() (NonlocalExit method) reshape() (AffineNormal method) (Gaussian method) (TorchDistributionMixin method) ReweightedWakeSleep (class in pyro.infer.rws) right_scale (AsymmetricLaplace property) (SoftAsymmetricLaplace property) RMSprop() (in module pyro.optim.pytorch_optimizers) Rprop() (in module pyro.optim.pytorch_optimizers) rsample() (AffineBeta method) (AffineNormal method) (AsymmetricLaplace method) (AVFMultivariateNormal method) (Delta method) (Gaussian method) (GaussianHMM method) (GaussianScaleMixture method) (IndependentHMM method) (LinearHMM method) (Logistic method) (MaskedDistribution method) (MaskedMixture method) (MixtureOfDiagNormals method) (MixtureOfDiagNormalsSharedCovariance method) (MultivariateStudentT method) (OMTMultivariateNormal method) (ProjectedNormal method) (Rejector method) (RelaxedBernoulliStraightThrough method) (RelaxedOneHotCategoricalStraightThrough method) (SkewLogistic method) (SoftAsymmetricLaplace method) (SoftLaplace method) (Stable method) (Unit method) rsample_posterior() (GaussianHMM method) run() (MCMC method) (StreamingMCMC method) (SVI method) (TracePosterior method) rv (Distribution property)

S

S() (Sylvester method) safe_cholesky() (in module pyro.ops.tensor_utils) safe_log() (in module pyro.ops.special) safe_normalize() (in module pyro.ops.tensor_utils) sample() (AffineBeta method) (BetaBinomial method) (CoalescentTimes method) (DirichletMultinomial method) (DiscreteHMM method) (Distribution method) (Empirical method) (GammaPoisson method) (Group method) (GroupedNormalNormal method) (HMC method) (ImproperUniform method) (in module pyro.contrib.autoname) (in module pyro.contrib.funsor) (in module pyro.contrib.minipyro) (in module pyro.primitives) (LogNormalNegativeBinomial method) (MaskedDistribution method) (MaskedMixture method) (MCMCKernel method) (MissingDataDiscreteHMM method) (NUTS method) (OneOneMatching method) (OneTwoMatching method) (RandomWalkKernel method) (Resampler method) (SineBivariateVonMises method) (SineSkewed method) (SpanningTree method) (TruncatedPolyaGamma method) (Unit method) (ZeroInflatedDistribution method) sample_() (Object method) sample_given_states() (MissingDataDiscreteHMM method) sample_latent() (AutoContinuous method) (AutoGuide method) sample_multi() (in module pyro.contrib.examples.multi_mnist) sample_one() (in module pyro.contrib.examples.multi_mnist) sample_posterior() (TraceEnum_ELBO method) sample_saved() (TraceEnumSample_ELBO method) sample_size (AffineBeta property) (Empirical property) sample_states() (MissingDataDiscreteHMM method) samples (WeighedPredictiveResults attribute) save() (ParamStoreDict method) (PyroOptim method) scale (AffineBeta property) (Message attribute) scale() (BlockMassMatrix method) (in module pyro.poutine.handlers) scale_constraint (AutoDiagonalNormal attribute) (AutoGaussian attribute) (AutoLowRankMultivariateNormal attribute) (AutoMultivariateNormal attribute) (AutoNormal attribute) (AutoStructured attribute) scale_tril (MultivariateStudentT property) scale_tril_constraint (AutoMultivariateNormal attribute) (AutoStructured attribute) ScaleMessenger (class in pyro.poutine.scale_messenger) scope() (in module pyro.contrib.autoname) (in module pyro.contrib.autoname.scoping) (ParamStoreDict method) ScopeMessenger (class in pyro.contrib.autoname.scoping) score_parts (Message attribute) score_parts() (Distribution method) (MaskedDistribution method) (Rejector method) seed (class in pyro.contrib.minipyro) seed() (in module pyro.poutine.handlers) SeedMessenger (class in pyro.poutine.seed_messenger) select_samples() (in module pyro.infer.mcmc.util) sequential_gaussian_filter_sample() (in module pyro.ops.gaussian) sequential_gaussian_tensordot() (in module pyro.ops.gaussian) SequentialLR() (in module pyro.optim.pytorch_optimizers) series (CompartmentalModel property) set() (in module pyro.settings) set_approx_log_prob_tol() (in module pyro.contrib.epidemiology.distributions) set_approx_sample_thresh() (in module pyro.contrib.epidemiology.distributions) set_data() (GPModel method) set_first_available_dim() (DimStack method) set_mode() (Parameterized method), [1] set_prior() (Parameterized method), [1] set_rng_seed() (in module pyro.contrib.funsor) set_state() (ParamStoreDict method) (PyroOptim method) setdefault() (ParamStoreDict method) setup() (HMC method) (MCMCKernel method) (RandomWalkKernel method) SGD() (in module pyro.optim.pytorch_optimizers) shape() (TorchDistributionMixin method) share_memory() (AdagradRMSProp method) sign (AffineAutoregressive attribute) (ELUTransform attribute) (LeakyReLUTransform attribute) (PositivePowerTransform attribute) (SoftplusTransform attribute) SimpleSEIRDModel (class in pyro.contrib.epidemiology.models) SimpleSEIRModel (class in pyro.contrib.epidemiology.models) SimpleSIRModel (class in pyro.contrib.epidemiology.models) SimplexToOrderedTransform (class in pyro.distributions.transforms) SineBivariateVonMises (class in pyro.distributions) SineSkewed (class in pyro.distributions) site_is_factor() (in module pyro.poutine.util) site_is_subsample() (in module pyro.poutine.util) size (CondIndepStackFrame attribute) SkewLogistic (class in pyro.distributions) SMCFailed

SMCFilter (class in pyro.infer.smcfilter) SMCState (class in pyro.infer.smcfilter) smooth() (MissingDataDiscreteHMM method) soft_scale (SoftAsymmetricLaplace property) SoftAsymmetricLaplace (class in pyro.distributions) SoftLaplace (class in pyro.distributions) SoftplusLowerCholeskyTransform (class in pyro.distributions.transforms) SoftplusTransform (class in pyro.distributions.transforms) SpanningTree (class in pyro.distributions) sparse_multinomial_likelihood() (in module pyro.ops.special) SparseAdam() (in module pyro.optim.pytorch_optimizers) SparseGPRegression (class in pyro.contrib.gp.models.sgpr) SparseSIRModel (class in pyro.contrib.epidemiology.models) Spline (class in pyro.distributions.transforms) spline() (in module pyro.distributions.transforms) spline_autoregressive() (in module pyro.distributions.transforms) spline_coupling() (in module pyro.distributions.transforms) SplineAutoregressive (class in pyro.distributions.transforms) SplineCoupling (class in pyro.distributions.transforms) split_gelman_rubin() (in module pyro.ops.stats) SplitReparam (class in pyro.infer.reparam.split) Stable (class in pyro.distributions) StableReparam (class in pyro.infer.reparam.stable) StableWithLogProb (class in pyro.distributions) StackFrame (class in pyro.contrib.funsor.handlers.runtime) StackStats (class in pyro.ops.streaming) StateDict (class in pyro.params.param_store) stationary_covariance() (MaternKernel method) StatsOfDict (class in pyro.ops.streaming) SteinKernel (class in pyro.infer.svgd) step() (AdagradRMSProp method) (ClippedAdam method) (DualAveraging method) (MixedMultiOptimizer method) (MultiOptimizer method) (PyroLRScheduler method) (PyroMultiOptimizer method) (SMCFilter method) (SVGD method) (SVI method), [1] step_size (HMC property) StepLR() (in module pyro.optim.pytorch_optimizers) stochastic_nodes (Trace property) stop (Message attribute) Strategy (class in pyro.infer.reparam.strategies) StreamingMCMC (class in pyro.infer.mcmc.api) StreamingStats (class in pyro.ops.streaming) StructuredReparam (class in pyro.infer.reparam.structured) StudentT (class in pyro.distributions) StudentTNet (class in pyro.contrib.cevae) StudentTReparam (class in pyro.infer.reparam.studentt) subsample() (in module pyro.contrib.funsor) (in module pyro.primitives) SubsampleMessenger (class in pyro.poutine.subsample_messenger) substitute() (in module pyro.poutine.handlers) SubstituteMessenger (class in pyro.poutine.substitute_messenger) successors() (Trace method) Sum (class in pyro.contrib.gp.kernels) summary() (MCMC method) SuperspreadingSEIRModel (class in pyro.contrib.epidemiology.models) SuperspreadingSIRModel (class in pyro.contrib.epidemiology.models) support (AffineBeta property) (AsymmetricLaplace attribute) (BetaBinomial property) (CoalescentTimes property) (CoalescentTimesWithRate property) (Delta property) (DirichletMultinomial property) (DiscreteHMM property) (Empirical attribute) (ExtendedBetaBinomial attribute) (ExtendedBinomial attribute) (FoldedDistribution attribute) (GammaGaussianHMM attribute) (GammaPoisson attribute) (GaussianHMM attribute) (GaussianMRF property) (GroupedNormalNormal attribute) (ImproperUniform property) (IndependentHMM property) (InverseGamma attribute) (LinearHMM property) (LKJ attribute) (Logistic attribute) (LogNormalNegativeBinomial attribute) (MaskedDistribution property) (MaskedMixture property) (MixtureOfDiagNormals attribute) (MixtureOfDiagNormalsSharedCovariance attribute) (MultivariateStudentT attribute) (OneOneMatching property) (OneTwoMatching property) (ProjectedNormal attribute) (SineBivariateVonMises attribute) (SineSkewed attribute) (SkewLogistic attribute) (SoftAsymmetricLaplace attribute) (SoftLaplace attribute) (SpanningTree attribute) (Stable attribute) (TruncatedPolyaGamma attribute) (Unit attribute) (VonMises3D attribute) (ZeroInflatedDistribution property) (ZeroInflatedNegativeBinomial attribute) (ZeroInflatedPoisson attribute) support() (Marginals method) SVGD (class in pyro.infer.svgd) SVI (class in pyro.contrib.minipyro) (class in pyro.infer.svi) Sylvester (class in pyro.distributions.transforms) sylvester() (in module pyro.distributions.transforms) symbolize_dims() (Trace method) SymmetricStableReparam (class in pyro.infer.reparam.stable)

T

t_dist() (Guide method) (Model method) terms_from_trace() (in module pyro.contrib.funsor.infer.traceenum_elbo) time (EKFState property) (Measurement property) time_plate (CompartmentalModel property) (ForecastingModel property) TimeSeriesModel (class in pyro.contrib.timeseries.base) tmc (InferDict attribute) to_data() (in module pyro.contrib.funsor) (in module pyro.contrib.funsor.handlers.primitives) to_event() (TorchDistributionMixin method) to_funsor() (in module pyro.contrib.funsor) (in module pyro.contrib.funsor.handlers.primitives) to_gaussian() (AffineNormal method) to_pyro_module_() (in module pyro.nn.module) to_script_module() (CEVAE method) topological_sort() (Trace method) TorchDistribution (class in pyro.distributions) TorchDistributionMixin (class in pyro.distributions.torch_distribution) TorchMultiOptimizer (class in pyro.optim.multi) total_count (ZeroInflatedNegativeBinomial property) trace (class in pyro.contrib.minipyro) Trace (class in pyro.poutine) trace (TraceHandler property) trace() (in module pyro.contrib.funsor.handlers) (in module pyro.ops.jit) (in module pyro.poutine.handlers) Trace_ELBO (class in pyro.contrib.funsor.infer.trace_elbo) (class in pyro.infer.trace_elbo) Trace_ELBO() (in module pyro.contrib.minipyro) TraceCausalEffect_ELBO (class in pyro.contrib.cevae) TraceEnum_ELBO (class in pyro.contrib.funsor.infer.traceenum_elbo) (class in pyro.infer.traceenum_elbo) TraceEnumSample_ELBO (class in pyro.infer.discrete) TraceGraph_ELBO (class in pyro.infer.tracegraph_elbo) TraceHandler (class in pyro.poutine.trace_messenger) TraceMarkovEnum_ELBO (class in pyro.contrib.funsor.infer.traceenum_elbo) TraceMeanField_ELBO (class in pyro.infer.trace_mean_field_elbo) TraceMessenger (class in pyro.poutine.trace_messenger) TracePosterior (class in pyro.infer.abstract_infer) TracePredictive (class in pyro.infer.abstract_infer) TraceTailAdaptive_ELBO (class in pyro.infer.trace_tail_adaptive_elbo) TraceTMC_ELBO (class in pyro.contrib.funsor.infer.tracetmc_elbo) (class in pyro.infer.tracetmc_elbo) TrackNonReparam (class in pyro.infer.tracegraph_elbo) train() (in module pyro.contrib.gp.util) training (AutoCallable attribute) (AutoContinuous attribute) (AutoDelta attribute) (AutoDiagonalNormal attribute) (AutoDiscreteParallel attribute) (AutoGaussian attribute) (AutoGuide attribute) (AutoGuideList attribute) (AutoHierarchicalNormalMessenger attribute) (AutoIAFNormal attribute) (AutoLaplaceApproximation attribute) (AutoLowRankMultivariateNormal attribute) (AutoMessenger attribute) (AutoMultivariateNormal attribute) (AutoNormal attribute) (AutoNormalizingFlow attribute) (AutoNormalMessenger attribute) (AutoRegressiveMessenger attribute) (AutoRegressiveNN attribute) (AutoStructured attribute) (BernoulliNet attribute) (Binary attribute) (Brownian attribute) (CEVAE attribute) (Combination attribute) (ConditionalAffineAutoregressive attribute) (ConditionalAffineCoupling attribute) (ConditionalAutoRegressiveNN attribute) (ConditionalDenseNN attribute) (ConditionalGeneralizedChannelPermute attribute) (ConditionalHouseholder attribute) (ConditionalMatrixExponential attribute) (ConditionalNeuralAutoregressive attribute) (ConditionalPlanar attribute) (ConditionalRadial attribute) (ConditionalSpline attribute) (ConditionalSplineAutoregressive attribute) (ConditionalTransformModule attribute) (Constant attribute) (Coregionalize attribute) (Cosine attribute) (DenseNN attribute) (DependentMaternGP attribute) (DiagNormalNet attribute) (DistributionNet attribute) (DotProduct attribute) (EasyGuide attribute) (ELBOModule attribute) (Exponent attribute) (Exponential attribute) (ExponentialNet attribute) (Gaussian attribute) (GenericLGSSM attribute) (GenericLGSSMWithGPNoiseModel attribute) (GPLVM attribute) (GPModel attribute) (GPRegression attribute) (Guide attribute) (IndependentMaternGP attribute) (Isotropy attribute) (Kernel attribute) (LaplaceNet attribute) (Likelihood attribute) (Linear attribute) (LinearlyCoupledMaternGP attribute) (Matern32 attribute) (Matern52 attribute) (MaternKernel attribute) (MHResampler attribute) (Model attribute) (MultiClass attribute) (NormalNet attribute) (Parameterized attribute), [1] (Periodic attribute) (Poisson attribute) (Polynomial attribute) (Predictive attribute) (Product attribute) (PyroModule attribute) (RationalQuadratic attribute) (RBF attribute) (SparseGPRegression attribute) (StudentTNet attribute) (Sum attribute) (Sylvester attribute) (TimeSeriesModel attribute) (Transforming attribute) (VariationalGP attribute) (VariationalSparseGP attribute) (VerticalScaling attribute) (Warping attribute) (WeighedPredictive attribute) (WhiteNoise attribute)

transform() (RandomVariable method) transform_sample() (NeuTraReparam method) transform_samples() (StructuredReparam method) TransformedDistribution (class in pyro.distributions) Transforming (class in pyro.contrib.gp.kernels) TransformModule (class in pyro.distributions) TransformReparam (class in pyro.infer.reparam.transform) transition() (CompartmentalModel method) transition_matrix() (MaternKernel method) transition_matrix_and_covariance() (MaternKernel method) triangular_solve() (in module pyro.ops.tensor_utils) TruncatedPolyaGamma (class in pyro.distributions) truncation_point (TruncatedPolyaGamma attribute) try_add() (ApproxSet method) type (Message attribute)

U

ubersum() (in module pyro.ops.contract) uncondition() (in module pyro.poutine.handlers) UnconditionMessenger (class in pyro.poutine.uncondition_messenger) unif01_goodness_of_fit() (in module pyro.distributions.testing.gof) Uniform (class in pyro.distributions) Unit (class in pyro.distributions) UnitJacobianReparam (class in pyro.infer.reparam.unit_jacobian) UnitLowerCholeskyTransform (class in pyro.distributions.transforms) UnknownStartSIRModel (class in pyro.contrib.epidemiology.models) unregister() (Messenger class method) unscale() (BlockMassMatrix method) unscaled_log_prob (Message attribute)

unwrap() (in module pyro.poutine.messenger) update() (BlockMassMatrix method) (CountMeanStats method) (CountMeanVarianceStats method) (CountStats method) (EKFState method) (StackStats method) (StatsOfDict method) (StreamingStats method) (WelfordArrowheadCovariance method) (WelfordCovariance method) upstream_value() (GuideMessenger method) user_param_name() (in module pyro.params.param_store)

V

validate_edges() (SpanningTree method) validation_enabled() (in module pyro.infer.util) (in module pyro.primitives) value (DimRequest attribute) (Message attribute) (ReparamMessage attribute) (ReparamResult attribute) values() (ParamStoreDict method) variance (AffineBeta property) (AsymmetricLaplace property) (BetaBinomial property) (Delta property) (DirichletMultinomial property) (Empirical property) (GammaPoisson property) (Logistic property) (LogNormalNegativeBinomial property) (MaskedDistribution property) (MaskedMixture property) (MultivariateStudentT property) (SoftAsymmetricLaplace property) (SoftLaplace property) (Stable property) (ZeroInflatedDistribution property)

VariationalGP (class in pyro.contrib.gp.models.vgp) VariationalSparseGP (class in pyro.contrib.gp.models.vsgp) vector_density_goodness_of_fit() (in module pyro.distributions.testing.gof) vectorize() (in module pyro.infer.svgd) vectorized (CondIndepStackFrame property) vectorized_importance_weights() (in module pyro.infer.importance) vectorized_markov() (in module pyro.contrib.funsor) (in module pyro.contrib.funsor.handlers) velocity_verlet() (in module pyro.ops.integrator) VerticalScaling (class in pyro.contrib.gp.kernels) vi_eig() (in module pyro.contrib.oed.eig) Vindex (class in pyro.ops.indexing) vindex() (in module pyro.ops.indexing) VISIBLE (DimType attribute) vnmc_eig() (in module pyro.contrib.oed.eig) volume_preserving (Householder attribute) (LowerCholeskyAffine attribute) (Permute attribute) VonMises (class in pyro.distributions) VonMises3D (class in pyro.distributions)

W

waic() (in module pyro.ops.stats) Warping (class in pyro.contrib.gp.kernels) was_observed (InferDict attribute) Weibull (class in pyro.distributions) weighed_quantile() (in module pyro.ops.stats) WeighedPredictive (class in pyro.infer.predictive) WeighedPredictiveResults (class in pyro.infer.predictive) weight_type (AutoHierarchicalNormalMessenger attribute) WelfordArrowheadCovariance (class in pyro.ops.welford) WelfordCovariance (class in pyro.ops.welford)

WhiteNoise (class in pyro.contrib.gp.kernels) Wishart (class in pyro.distributions) with_cache() (ComposeTransformModule method) (DiscreteCosineTransform method) (HaarTransform method) (LowerCholeskyAffine method) (Normalize method) (Permute method) (PositivePowerTransform method) write() (in module pyro.contrib.mue.dataloaders)

X

x_dist() (Model method)

Y

y_dist() (Guide method) (Model method)

y_mean() (Model method)

Z

z_dist() (Guide method) (Model method) ZeroInflatedDistribution (class in pyro.distributions)

ZeroInflatedNegativeBinomial (class in pyro.distributions) ZeroInflatedPoisson (class in pyro.distributions) ZukoToPyro (class in pyro.contrib.zuko)