Path	Lines of Code
examples/air/air.py	227
examples/air/main.py	246
examples/air/modules.py	57
examples/air/viz.py	36
examples/baseball.py	209
examples/capture_recapture/cjs.py	170
examples/contrib/autoname/mixture.py	52
examples/contrib/autoname/scoping_mixture.py	51
examples/contrib/autoname/tree_data.py	81
examples/contrib/cevae/synthetic.py	64
examples/contrib/epidemiology/regional.py	159
examples/contrib/epidemiology/sir.py	293
examples/contrib/forecast/bart.py	106
examples/contrib/gp/sv-dkl.py	129
examples/contrib/oed/gp_bayes_opt.py	71
examples/contrib/timeseries/gp_models.py	112
examples/dmm/__init__.py	1
examples/dmm/dmm.py	266
examples/dmm/polyphonic_data_loader.py	97
examples/dmm/util.py	9
examples/eight_schools/data.py	4
examples/eight_schools/mcmc.py	39
examples/eight_schools/svi.py	61
examples/einsum.py	104
examples/hmm.py	343
examples/inclined_plane.py	66
examples/lda.py	96
examples/lkj.py	40
examples/minipyro.py	38
examples/mixed_hmm/__init__.py	1
examples/mixed_hmm/experiment.py	117
examples/mixed_hmm/model.py	132
examples/mixed_hmm/seal_data.py	43
examples/neutra.py	155
examples/rsa/generics.py	108
examples/rsa/hyperbole.py	104
examples/rsa/schelling.py	41
examples/rsa/schelling_false.py	45
examples/rsa/search_inference.py	136
examples/rsa/semantic_parsing.py	241
examples/sir_hmc.py	350
examples/smcfilter.py	75
examples/sparse_gamma_def.py	149
examples/sparse_regression.py	191
examples/svi_horovod.py	94
examples/toy_mixture_model_discrete_enumeration.py	92
examples/vae/ss_vae_M2.py	219
examples/vae/utils/__init__.py	1
examples/vae/utils/custom_mlp.py	81
examples/vae/utils/mnist_cached.py	135
examples/vae/utils/vae_plots.py	73
examples/vae/vae.py	122
examples/vae/vae_comparison.py	179
profiler/__init__.py	1
profiler/distributions.py	149
profiler/hmm.py	60
profiler/profiling_utils.py	95
pyro/__init__.py	37
pyro/contrib/__init__.py	18
pyro/contrib/autoguide.py	5
pyro/contrib/autoname/__init__.py	7
pyro/contrib/autoname/named.py	104
pyro/contrib/autoname/scoping.py	57
pyro/contrib/bnn/__init__.py	4
pyro/contrib/bnn/hidden_layer.py	49
pyro/contrib/bnn/utils.py	10
pyro/contrib/cevae/__init__.py	291
pyro/contrib/conjugate/__init__.py	1
pyro/contrib/conjugate/infer.py	137
pyro/contrib/easyguide/__init__.py	5
pyro/contrib/easyguide/easyguide.py	162
pyro/contrib/epidemiology/__init__.py	10
pyro/contrib/epidemiology/compartmental.py	621
pyro/contrib/epidemiology/distributions.py	135
pyro/contrib/epidemiology/models.py	554
pyro/contrib/epidemiology/util.py	173
pyro/contrib/examples/__init__.py	1
pyro/contrib/examples/bart.py	127
pyro/contrib/examples/finance.py	13
pyro/contrib/examples/multi_mnist.py	63
pyro/contrib/examples/util.py	60
pyro/contrib/forecast/__init__.py	11
pyro/contrib/forecast/evaluate.py	122
pyro/contrib/forecast/forecaster.py	207
pyro/contrib/forecast/util.py	238
pyro/contrib/funsor/__init__.py	18
pyro/contrib/funsor/handlers/__init__.py	22
pyro/contrib/funsor/handlers/enum_messenger.py	136
pyro/contrib/funsor/handlers/named_messenger.py	113
pyro/contrib/funsor/handlers/plate_messenger.py	111
pyro/contrib/funsor/handlers/primitives.py	16
pyro/contrib/funsor/handlers/replay_messenger.py	24
pyro/contrib/funsor/handlers/runtime.py	129
pyro/contrib/funsor/handlers/trace_messenger.py	32
pyro/contrib/funsor/infer/__init__.py	5
pyro/contrib/funsor/infer/elbo.py	29
pyro/contrib/funsor/infer/trace_elbo.py	29
pyro/contrib/funsor/infer/traceenum_elbo.py	63
pyro/contrib/funsor/infer/tracetmc_elbo.py	31
pyro/contrib/gp/__init__.py	10
pyro/contrib/gp/kernels/__init__.py	47
pyro/contrib/gp/kernels/brownian.py	25
pyro/contrib/gp/kernels/coregionalize.py	35
pyro/contrib/gp/kernels/dot_product.py	38
pyro/contrib/gp/kernels/isotropic.py	85
pyro/contrib/gp/kernels/kernel.py	101
pyro/contrib/gp/kernels/periodic.py	37
pyro/contrib/gp/kernels/static.py	29
pyro/contrib/gp/likelihoods/__init__.py	25
pyro/contrib/gp/likelihoods/binary.py	19
pyro/contrib/gp/likelihoods/gaussian.py	18
pyro/contrib/gp/likelihoods/likelihood.py	6
pyro/contrib/gp/likelihoods/multi_class.py	31
pyro/contrib/gp/likelihoods/poisson.py	16
pyro/contrib/gp/models/__init__.py	14
pyro/contrib/gp/models/gplvm.py	27
pyro/contrib/gp/models/gpr.py	89
pyro/contrib/gp/models/model.py	35
pyro/contrib/gp/models/sgpr.py	106
pyro/contrib/gp/models/vgp.py	66
pyro/contrib/gp/models/vsgp.py	68
pyro/contrib/gp/parameterized.py	103
pyro/contrib/gp/util.py	74
pyro/contrib/minipyro.py	210
pyro/contrib/oed/__init__.py	5
pyro/contrib/oed/eig.py	404
pyro/contrib/oed/glmm/__init__.py	2
pyro/contrib/oed/glmm/glmm.py	211
pyro/contrib/oed/glmm/guides.py	156
pyro/contrib/oed/search.py	16
pyro/contrib/oed/util.py	21
pyro/contrib/randomvariable/__init__.py	4
pyro/contrib/randomvariable/random_variable.py	68
pyro/contrib/timeseries/__init__.py	12
pyro/contrib/timeseries/base.py	10
pyro/contrib/timeseries/gp.py	238
pyro/contrib/timeseries/lgssm.py	68
pyro/contrib/timeseries/lgssmgp.py	106
pyro/contrib/tracking/__init__.py	4
pyro/contrib/tracking/assignment.py	194
pyro/contrib/tracking/distributions.py	49
pyro/contrib/tracking/dynamic_models.py	364
pyro/contrib/tracking/extended_kalman_filter.py	176
pyro/contrib/tracking/hashing.py	89
pyro/contrib/tracking/measurements.py	116
pyro/contrib/util.py	44
pyro/distributions/__init__.py	93
pyro/distributions/avf_mvn.py	47
pyro/distributions/coalescent.py	231
pyro/distributions/conditional.py	43
pyro/distributions/conjugate.py	158
pyro/distributions/constraints.py	31
pyro/distributions/delta.py	45
pyro/distributions/diag_normal_mixture.py	147
pyro/distributions/diag_normal_mixture_shared_cov.py	118
pyro/distributions/distribution.py	33
pyro/distributions/empirical.py	79
pyro/distributions/extended.py	31
pyro/distributions/folded.py	18
pyro/distributions/gaussian_scale_mixture.py	94
pyro/distributions/hmm.py	579
pyro/distributions/improper_uniform.py	25
pyro/distributions/inverse_gamma.py	21
pyro/distributions/kl.py	43
pyro/distributions/lkj.py	68
pyro/distributions/mixture.py	90
pyro/distributions/multivariate_studentt.py	85
pyro/distributions/omt_mvn.py	51
pyro/distributions/polya_gamma.py	37
pyro/distributions/rejector.py	46
pyro/distributions/relaxed_straight_through.py	45
pyro/distributions/score_parts.py	8
pyro/distributions/spanning_tree.py	273
pyro/distributions/stable.py	88
pyro/distributions/testing/__init__.py	1
pyro/distributions/testing/fakes.py	9
pyro/distributions/testing/naive_dirichlet.py	21
pyro/distributions/testing/rejection_exponential.py	24
pyro/distributions/testing/rejection_gamma.py	147
pyro/distributions/torch.py	168
pyro/distributions/torch_distribution.py	218
pyro/distributions/torch_patch.py	52
pyro/distributions/torch_transform.py	13
pyro/distributions/transforms/__init__.py	113
pyro/distributions/transforms/affine_autoregressive.py	119
pyro/distributions/transforms/affine_coupling.py	108
pyro/distributions/transforms/basic.py	39
pyro/distributions/transforms/batchnorm.py	46
pyro/distributions/transforms/block_autoregressive.py	112
pyro/distributions/transforms/cholesky.py	51
pyro/distributions/transforms/discrete_cosine.py	52
pyro/distributions/transforms/generalized_channel_permute.py	87
pyro/distributions/transforms/haar.py	41
pyro/distributions/transforms/householder.py	99
pyro/distributions/transforms/lower_cholesky_affine.py	29
pyro/distributions/transforms/matrix_exponential.py	89
pyro/distributions/transforms/neural_autoregressive.py	91
pyro/distributions/transforms/permute.py	43
pyro/distributions/transforms/planar.py	83
pyro/distributions/transforms/polynomial.py	57
pyro/distributions/transforms/radial.py	86
pyro/distributions/transforms/spline.py	261
pyro/distributions/transforms/spline_autoregressive.py	77
pyro/distributions/transforms/spline_coupling.py	64
pyro/distributions/transforms/sylvester.py	62
pyro/distributions/transforms/utils.py	2
pyro/distributions/unit.py	24
pyro/distributions/util.py	147
pyro/distributions/von_mises_3d.py	31
pyro/distributions/zero_inflated.py	89
pyro/generic.py	4
pyro/infer/__init__.py	59
pyro/infer/abstract_infer.py	187
pyro/infer/autoguide/__init__.py	30
pyro/infer/autoguide/guides.py	493
pyro/infer/autoguide/initialization.py	96
pyro/infer/autoguide/utils.py	15
pyro/infer/csis.py	78
pyro/infer/discrete.py	127
pyro/infer/elbo.py	79
pyro/infer/energy_distance.py	126
pyro/infer/enum.py	90
pyro/infer/importance.py	105
pyro/infer/mcmc/__init__.py	11
pyro/infer/mcmc/adaptation.py	299
pyro/infer/mcmc/api.py	311
pyro/infer/mcmc/hmc.py	255
pyro/infer/mcmc/logger.py	125
pyro/infer/mcmc/mcmc_kernel.py	24
pyro/infer/mcmc/nuts.py	227
pyro/infer/mcmc/util.py	380
pyro/infer/predictive.py	129
pyro/infer/renyi_elbo.py	122
pyro/infer/reparam/__init__.py	26
pyro/infer/reparam/conjugate.py	23
pyro/infer/reparam/discrete_cosine.py	6
pyro/infer/reparam/haar.py	6
pyro/infer/reparam/hmm.py	49
pyro/infer/reparam/loc_scale.py	42
pyro/infer/reparam/neutra.py	54
pyro/infer/reparam/reparam.py	16
pyro/infer/reparam/split.py	27
pyro/infer/reparam/stable.py	96
pyro/infer/reparam/studentt.py	14
pyro/infer/reparam/transform.py	12
pyro/infer/reparam/unit_jacobian.py	23
pyro/infer/rws.py	116
pyro/infer/smcfilter.py	112
pyro/infer/svgd.py	141
pyro/infer/svi.py	80
pyro/infer/trace_elbo.py	123
pyro/infer/trace_mean_field_elbo.py	92
pyro/infer/trace_mmd.py	122
pyro/infer/trace_tail_adaptive_elbo.py	30
pyro/infer/traceenum_elbo.py	334
pyro/infer/tracegraph_elbo.py	183
pyro/infer/tracetmc_elbo.py	118
pyro/infer/util.py	176
pyro/logger.py	10
pyro/nn/__init__.py	15
pyro/nn/auto_reg_nn.py	130
pyro/nn/dense_nn.py	58
pyro/nn/module.py	311
pyro/ops/__init__.py	1
pyro/ops/arrowhead.py	61
pyro/ops/contract.py	268
pyro/ops/dual_averaging.py	20
pyro/ops/einsum/__init__.py	22
pyro/ops/einsum/adjoint.py	102
pyro/ops/einsum/torch_log.py	34
pyro/ops/einsum/torch_map.py	36
pyro/ops/einsum/torch_marginal.py	44
pyro/ops/einsum/torch_sample.py	47
pyro/ops/einsum/util.py	33
pyro/ops/gamma_gaussian.py	216
pyro/ops/gaussian.py	297
pyro/ops/hessian.py	10
pyro/ops/indexing.py	69
pyro/ops/integrator.py	41
pyro/ops/jit.py	86
pyro/ops/linalg.py	56
pyro/ops/newton.py	80
pyro/ops/packed.py	116
pyro/ops/rings.py	186
pyro/ops/special.py	37
pyro/ops/ssm_gp.py	88
pyro/ops/stats.py	187
pyro/ops/tensor_utils.py	188
pyro/ops/welford.py	62
pyro/optim/__init__.py	15
pyro/optim/adagrad_rmsprop.py	38
pyro/optim/clipped_adam.py	39
pyro/optim/dct_adam.py	109
pyro/optim/horovod.py	20
pyro/optim/lr_scheduler.py	16
pyro/optim/multi.py	58
pyro/optim/optim.py	93
pyro/optim/pytorch_optimizers.py	35
pyro/params/__init__.py	6
pyro/params/param_store.py	109
pyro/poutine/__init__.py	28
pyro/poutine/block_messenger.py	58
pyro/poutine/broadcast_messenger.py	28
pyro/poutine/condition_messenger.py	15
pyro/poutine/do_messenger.py	31
pyro/poutine/enum_messenger.py	140
pyro/poutine/escape_messenger.py	14
pyro/poutine/handlers.py	94
pyro/poutine/indep_messenger.py	73
pyro/poutine/infer_config_messenger.py	11
pyro/poutine/lift_messenger.py	66
pyro/poutine/markov_messenger.py	47
pyro/poutine/mask_messenger.py	18
pyro/poutine/messenger.py	69
pyro/poutine/plate_messenger.py	11
pyro/poutine/reentrant_messenger.py	17
pyro/poutine/reparam_messenger.py	39
pyro/poutine/replay_messenger.py	29
pyro/poutine/runtime.py	120
pyro/poutine/scale_messenger.py	16
pyro/poutine/seed_messenger.py	12
pyro/poutine/subsample_messenger.py	117
pyro/poutine/trace_messenger.py	92
pyro/poutine/trace_struct.py	277
pyro/poutine/uncondition_messenger.py	12
pyro/poutine/util.py	49
pyro/primitives.py	141
pyro/util.py	300
scripts/update_headers.py	57
setup.py	102
tutorial/source/cleannb.py	19
tutorial/source/conf.py	49
tutorial/source/contrib_funsor_intro_i.ipynb	674
tutorial/source/contrib_funsor_intro_ii.ipynb	503
tutorial/source/custom_objectives.ipynb	255
tutorial/source/easyguide.ipynb	322
tutorial/source/effect_handlers.ipynb	725
tutorial/source/ekf.ipynb	265
tutorial/source/elections.ipynb	807
tutorial/source/enumeration.ipynb	645
tutorial/source/intro_part_i.ipynb	322
tutorial/source/jit.ipynb	401
tutorial/source/mle_map.ipynb	282
tutorial/source/modules.ipynb	763
tutorial/source/search_inference.py	142
tutorial/source/ss-vae.ipynb	538
tutorial/source/svi_part_i.ipynb	387
tutorial/source/svi_part_ii.ipynb	235
tutorial/source/tensor_shapes.ipynb	769
tutorial/source/vae.ipynb	662