README.md
setup.cfg
setup.py
numpyro/__init__.py
numpyro/diagnostics.py
numpyro/handlers.py
numpyro/optim.py
numpyro/patch.py
numpyro/primitives.py
numpyro/util.py
numpyro/version.py
numpyro.egg-info/PKG-INFO
numpyro.egg-info/SOURCES.txt
numpyro.egg-info/dependency_links.txt
numpyro.egg-info/requires.txt
numpyro.egg-info/top_level.txt
numpyro/compat/__init__.py
numpyro/compat/distributions.py
numpyro/compat/handlers.py
numpyro/compat/infer.py
numpyro/compat/ops.py
numpyro/compat/optim.py
numpyro/compat/pyro.py
numpyro/compat/util.py
numpyro/contrib/__init__.py
numpyro/contrib/autoguide.py
numpyro/contrib/indexing.py
numpyro/contrib/control_flow/__init__.py
numpyro/contrib/control_flow/scan.py
numpyro/contrib/funsor/__init__.py
numpyro/contrib/funsor/enum_messenger.py
numpyro/contrib/funsor/infer_util.py
numpyro/distributions/__init__.py
numpyro/distributions/conjugate.py
numpyro/distributions/constraints.py
numpyro/distributions/continuous.py
numpyro/distributions/directional.py
numpyro/distributions/discrete.py
numpyro/distributions/distribution.py
numpyro/distributions/flows.py
numpyro/distributions/transforms.py
numpyro/distributions/util.py
numpyro/examples/__init__.py
numpyro/examples/datasets.py
numpyro/infer/__init__.py
numpyro/infer/autoguide.py
numpyro/infer/elbo.py
numpyro/infer/hmc.py
numpyro/infer/hmc_util.py
numpyro/infer/initialization.py
numpyro/infer/mcmc.py
numpyro/infer/reparam.py
numpyro/infer/sa.py
numpyro/infer/svi.py
numpyro/infer/util.py
numpyro/nn/__init__.py
numpyro/nn/auto_reg_nn.py
numpyro/nn/block_neural_arn.py
numpyro/nn/masked_dense.py
test/test_autoguide.py
test/test_compile.py
test/test_diagnostics.py
test/test_distributions.py
test/test_distributions_util.py
test/test_example_utils.py
test/test_examples.py
test/test_flows.py
test/test_handlers.py
test/test_hmc_util.py
test/test_infer_util.py
test/test_mcmc.py
test/test_nn.py
test/test_optimizers.py
test/test_reparam.py
test/test_svi.py
test/test_util.py