LICENSE.md
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/module.py
numpyro/contrib/nested_sampling.py
numpyro/contrib/render.py
numpyro/contrib/control_flow/__init__.py
numpyro/contrib/control_flow/cond.py
numpyro/contrib/control_flow/scan.py
numpyro/contrib/einstein/__init__.py
numpyro/contrib/einstein/kernels.py
numpyro/contrib/einstein/steinvi.py
numpyro/contrib/einstein/util.py
numpyro/contrib/funsor/__init__.py
numpyro/contrib/funsor/discrete.py
numpyro/contrib/funsor/enum_messenger.py
numpyro/contrib/funsor/infer_util.py
numpyro/contrib/tfp/__init__.py
numpyro/contrib/tfp/distributions.py
numpyro/contrib/tfp/mcmc.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/gof.py
numpyro/distributions/kl.py
numpyro/distributions/mixtures.py
numpyro/distributions/transforms.py
numpyro/distributions/truncated.py
numpyro/distributions/util.py
numpyro/examples/__init__.py
numpyro/examples/datasets.py
numpyro/infer/__init__.py
numpyro/infer/autoguide.py
numpyro/infer/barker.py
numpyro/infer/elbo.py
numpyro/infer/hmc.py
numpyro/infer/hmc_gibbs.py
numpyro/infer/hmc_util.py
numpyro/infer/initialization.py
numpyro/infer/inspect.py
numpyro/infer/mcmc.py
numpyro/infer/mixed_hmc.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
numpyro/ops/__init__.py
numpyro/ops/indexing.py
numpyro/ops/provenance.py
numpyro/ops/pytree.py
test/test_compile.py
test/test_diagnostics.py
test/test_distributions.py
test/test_distributions_mixture.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_model_rendering.py
test/test_nn.py
test/test_optimizers.py
test/test_pickle.py
test/test_util.py