Upload 883 files

.gitattributes

@@ -33,3 +33,11 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+pyro/source/tutorial/source/_static/img/first_embedding.png filter=lfs diff=lfs merge=lfs -text
+pyro/source/tutorial/source/_static/img/gpr-fit.gif filter=lfs diff=lfs merge=lfs -text
+pyro/source/tutorial/source/_static/img/guide.png filter=lfs diff=lfs merge=lfs -text
+pyro/source/tutorial/source/_static/img/model.png filter=lfs diff=lfs merge=lfs -text
+pyro/source/tutorial/source/_static/img/sir_hmc/forecast.png filter=lfs diff=lfs merge=lfs -text
+pyro/source/tutorial/source/_static/img/ss_vae_zoo.png filter=lfs diff=lfs merge=lfs -text
+pyro/source/tutorial/source/_static/img/svgpr-fit.gif filter=lfs diff=lfs merge=lfs -text
+pyro/source/tutorial/source/_static/img/third_embedding.png filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,18 @@
+FROM python:3.10
+
+RUN useradd -m -u 1000 user && python -m pip install --upgrade pip
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+ENV MCP_TRANSPORT=http
+ENV MCP_PORT=7860
+
+EXPOSE 7860
+
+CMD ["python", "pyro/mcp_output/start_mcp.py"]

README.md

@@ -1,10 +1,32 @@
 ---
-title: Pyro
-emoji: 🌍
-colorFrom: indigo
-colorTo: indigo
+title: Pyro MCP
+emoji: 🤖
+colorFrom: blue
+colorTo: purple
 sdk: docker
+sdk_version: "4.26.0"
+app_file: app.py
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Pyro MCP Service
+
+Auto-generated MCP service for pyro.
+
+## Usage
+
+```
+https://None-pyro-mcp.hf.space/mcp
+```
+
+## Connect with Cursor
+
+```json
+{
+  "mcpServers": {
+    "pyro": {
+      "url": "https://None-pyro-mcp.hf.space/mcp"
+    }
+  }
+}
+```

app.py

@@ -0,0 +1,45 @@
+from fastapi import FastAPI
+import os
+import sys
+
+mcp_plugin_path = os.path.join(os.path.dirname(__file__), "pyro", "mcp_output", "mcp_plugin")
+sys.path.insert(0, mcp_plugin_path)
+
+app = FastAPI(
+    title="Pyro MCP Service",
+    description="Auto-generated MCP service for pyro",
+    version="1.0.0"
+)
+
+@app.get("/")
+def root():
+    return {
+        "service": "Pyro MCP Service",
+        "version": "1.0.0",
+        "status": "running",
+        "transport": os.environ.get("MCP_TRANSPORT", "http")
+    }
+
+@app.get("/health")
+def health_check():
+    return {"status": "healthy", "service": "pyro MCP"}
+
+@app.get("/tools")
+def list_tools():
+    try:
+        from mcp_service import create_app
+        mcp_app = create_app()
+        tools = []
+        for tool_name, tool_func in mcp_app.tools.items():
+            tools.append({
+                "name": tool_name,
+                "description": tool_func.__doc__ or "No description available"
+            })
+        return {"tools": tools}
+    except Exception as e:
+        return {"error": f"Failed to load tools: {str(e)}"}
+
+if __name__ == "__main__":
+    import uvicorn
+    port = int(os.environ.get("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)

pyro/mcp_output/README_MCP.md

@@ -0,0 +1,59 @@
+# Pyro: Probabilistic Programming with PyTorch
+
+## Project Introduction
+
+Pyro is a universal probabilistic programming language (PPL) built on PyTorch. It allows developers to write generative models as Python programs and perform inference over them using a variety of algorithms, including Stochastic Variational Inference (SVI) and Markov Chain Monte Carlo (MCMC). Pyro is designed to be flexible, scalable, and minimal, providing powerful abstractions for probabilistic modeling and inference.
+
+## Installation Method
+
+To install Pyro, ensure you have Python 3.8 or higher. You can install Pyro and its core dependencies using pip:
+
+```
+pip install pyro-ppl
+```
+
+Pyro requires `torch` and `numpy` as core dependencies. Optional dependencies include `matplotlib` and `scipy` for enhanced functionality.
+
+## Quick Start
+
+Here's a simple example to get you started with Pyro:
+
+1. Import Pyro and necessary modules.
+2. Define a model using Pyro's primitives like `sample` and `param`.
+3. Perform inference using SVI or MCMC.
+
+Example:
+
+```
+import pyro
+import pyro.distributions as dist
+
+def model(data):
+    alpha = pyro.param("alpha", torch.tensor(1.0))
+    beta = pyro.param("beta", torch.tensor(1.0))
+    with pyro.plate("data", len(data)):
+        pyro.sample("obs", dist.Beta(alpha, beta), obs=data)
+
+# Perform inference using SVI or MCMC
+```
+
+## Available Tools and Endpoints List
+
+- **Core Primitives**: Functions like `sample`, `param`, `plate`, `factor`, and `deterministic` for building probabilistic models.
+- **Effect Handlers (Poutine)**: Tools like `trace`, `replay`, `condition`, `block`, `scale`, and `enum` for intercepting and transforming probabilistic primitives.
+- **Inference Algorithms**: Includes `SVI`, `MCMC`, and `Importance` for performing inference on models.
+- **Distributions**: Extensions to PyTorch's distribution library with additional distributions like `Delta`, `HiddenMarkovModel`, and `Empirical`.
+
+## Common Issues and Notes
+
+- **Dependencies**: Ensure that `torch` and `numpy` are installed. Optional dependencies like `matplotlib` and `scipy` can enhance functionality but are not required.
+- **Environment**: Pyro requires Python 3.8 or higher. Ensure your environment is set up accordingly.
+- **Performance**: Pyro is designed to scale with minimal overhead. However, performance can be affected by the complexity of the model and the size of the data.
+
+## Reference Links or Documentation
+
+- [Pyro GitHub Repository](https://github.com/pyro-ppl/pyro)
+- [Pyro Documentation](https://pyro.ai/)
+- [Pyro Tutorials](https://pyro.ai/examples/)
+
+For more detailed information on Pyro's architecture, design philosophy, and core components, please refer to the official documentation and tutorials.

pyro/mcp_output/analysis.json

@@ -0,0 +1,2110 @@
+{
+  "summary": {
+    "repository_url": "https://github.com/pyro-ppl/pyro",
+    "summary": "Imported via zip fallback, file count: 655",
+    "file_tree": {
+      ".codecov.yml": {
+        "size": 126
+      },
+      ".github/FUNDING.yml": {
+        "size": 812
+      },
+      ".github/issue_template.md": {
+        "size": 975
+      },
+      ".github/workflows/ci.yml": {
+        "size": 10372
+      },
+      ".github/workflows/publish.yml": {
+        "size": 990
+      },
+      ".readthedocs.yml": {
+        "size": 413
+      },
+      "CODE_OF_CONDUCT.md": {
+        "size": 3212
+      },
+      "CONTRIBUTING.md": {
+        "size": 3956
+      },
+      "LICENSE.md": {
+        "size": 11358
+      },
+      "README.md": {
+        "size": 4538
+      },
+      "RELEASE-MANAGEMENT.md": {
+        "size": 1628
+      },
+      "docker/README.md": {
+        "size": 2649
+      },
+      "docs/README.md": {
+        "size": 545
+      },
+      "docs/requirements.txt": {
+        "size": 153
+      },
+      "docs/source/_static/img/favicon/manifest.json": {
+        "size": 720
+      },
+      "docs/source/conf.py": {
+        "size": 6997
+      },
+      "docs/source/pyro.infer.mcmc.txt": {
+        "size": 1053
+      },
+      "docs/source/pyro.optim.txt": {
+        "size": 1181
+      },
+      "docs/source/pyro.poutine.txt": {
+        "size": 3375
+      },
+      "examples/__init__.py": {
+        "size": 0
+      },
+      "examples/air/air.py": {
+        "size": 14031
+      },
+      "examples/air/main.py": {
+        "size": 14628
+      },
+      "examples/air/modules.py": {
+        "size": 2991
+      },
+      "examples/air/viz.py": {
+        "size": 2528
+      },
+      "examples/baseball.py": {
+        "size": 16383
+      },
+      "examples/capture_recapture/cjs.py": {
+        "size": 14374
+      },
+      "examples/contrib/__init__.py": {
+        "size": 0
+      },
+      "examples/contrib/autoname/mixture.py": {
+        "size": 2572
+      },
+      "examples/contrib/autoname/scoping_mixture.py": {
+        "size": 2239
+      },
+      "examples/contrib/autoname/tree_data.py": {
+        "size": 3560
+      },
+      "examples/contrib/cevae/synthetic.py": {
+        "size": 3802
+      },
+      "examples/contrib/epidemiology/regional.py": {
+        "size": 7900
+      },
+      "examples/contrib/epidemiology/sir.py": {
+        "size": 15144
+      },
+      "examples/contrib/forecast/bart.py": {
+        "size": 7530
+      },
+      "examples/contrib/funsor/__init__.py": {
+        "size": 0
+      },
+      "examples/contrib/funsor/hmm.py": {
+        "size": 35805
+      },
+      "examples/contrib/gp/sv-dkl.py": {
+        "size": 8529
+      },
+      "examples/contrib/mue/FactorMuE.py": {
+        "size": 13776
+      },
+      "examples/contrib/mue/ProfileHMM.py": {
+        "size": 10356
+      },
+      "examples/contrib/oed/ab_test.py": {
+        "size": 4658
+      },
+      "examples/contrib/oed/gp_bayes_opt.py": {
+        "size": 5456
+      },
+      "examples/contrib/timeseries/gp_models.py": {
+        "size": 7165
+      },
+      "examples/cvae/__init__.py": {
+        "size": 0
+      },
+      "examples/cvae/baseline.py": {
+        "size": 3456
+      },
+      "examples/cvae/cvae.py": {
+        "size": 6875
+      },
+      "examples/cvae/main.py": {
+        "size": 4147
+      },
+      "examples/cvae/mnist.py": {
+        "size": 3204
+      },
+      "examples/cvae/util.py": {
+        "size": 4845
+      },
+      "examples/dmm.py": {
+        "size": 25355
+      },
+      "examples/eight_schools/README.md": {
+        "size": 672
+      },
+      "examples/eight_schools/data.py": {
+        "size": 249
+      },
+      "examples/eight_schools/mcmc.py": {
+        "size": 1818
+      },
+      "examples/eight_schools/svi.py": {
+        "size": 3025
+      },
+      "examples/einsum.py": {
+        "size": 7580
+      },
+      "examples/hmm.py": {
+        "size": 32364
+      },
+      "examples/inclined_plane.py": {
+        "size": 5455
+      },
+      "examples/lda.py": {
+        "size": 6708
+      },
+      "examples/lkj.py": {
+        "size": 2735
+      },
+      "examples/minipyro.py": {
+        "size": 2971
+      },
+      "examples/mixed_hmm/README.md": {
+        "size": 4517
+      },
+      "examples/mixed_hmm/__init__.py": {
+        "size": 0
+      },
+      "examples/mixed_hmm/experiment.py": {
+        "size": 5959
+      },
+      "examples/mixed_hmm/model.py": {
+        "size": 9884
+      },
+      "examples/mixed_hmm/seal_data.py": {
+        "size": 2549
+      },
+      "examples/neutra.py": {
+        "size": 9381
+      },
+      "examples/rsa/README.md": {
+        "size": 972
+      },
+      "examples/rsa/generics.py": {
+        "size": 5389
+      },
+      "examples/rsa/hyperbole.py": {
+        "size": 6737
+      },
+      "examples/rsa/schelling.py": {
+        "size": 2839
+      },
+      "examples/rsa/schelling_false.py": {
+        "size": 3334
+      },
+      "examples/rsa/search_inference.py": {
+        "size": 7562
+      },
+      "examples/rsa/semantic_parsing.py": {
+        "size": 8669
+      },
+      "examples/scanvi/__init__.py": {
+        "size": 0
+      },
+      "examples/scanvi/scanvi.py": {
+        "size": 16796
+      },
+      "examples/sir_hmc.py": {
+        "size": 24193
+      },
+      "examples/smcfilter.py": {
+        "size": 3358
+      },
+      "examples/sparse_gamma_def.py": {
+        "size": 12095
+      },
+      "examples/sparse_regression.py": {
+        "size": 13613
+      },
+      "examples/svi_horovod.py": {
+        "size": 6668
+      },
+      "examples/svi_lightning.py": {
+        "size": 4919
+      },
+      "examples/svi_torch.py": {
+        "size": 5054
+      },
+      "examples/toy_mixture_model_discrete_enumeration.py": {
+        "size": 5197
+      },
+      "examples/vae/ss_vae_M2.py": {
+        "size": 20428
+      },
+      "examples/vae/utils/__init__.py": {
+        "size": 0
+      },
+      "examples/vae/utils/custom_mlp.py": {
+        "size": 6871
+      },
+      "examples/vae/utils/mnist_cached.py": {
+        "size": 8857
+      },
+      "examples/vae/utils/vae_plots.py": {
+        "size": 3678
+      },
+      "examples/vae/vae.py": {
+        "size": 9201
+      },
+      "examples/vae/vae_comparison.py": {
+        "size": 9094
+      },
+      "profiler/__init__.py": {
+        "size": 0
+      },
+      "profiler/distributions.py": {
+        "size": 5360
+      },
+      "profiler/gaussianhmm.py": {
+        "size": 2678
+      },
+      "profiler/hmm.py": {
+        "size": 2834
+      },
+      "profiler/profiling_utils.py": {
+        "size": 4006
+      },
+      "pyproject.toml": {
+        "size": 100
+      },
+      "pyro/__init__.py": {
+        "size": 1324
+      },
+      "pyro/contrib/README.md": {
+        "size": 608
+      },
+      "pyro/contrib/__init__.py": {
+        "size": 693
+      },
+      "pyro/contrib/autoguide.py": {
+        "size": 317
+      },
+      "pyro/contrib/autoname/__init__.py": {
+        "size": 486
+      },
+      "pyro/contrib/autoname/autoname.py": {
+        "size": 5100
+      },
+      "pyro/contrib/autoname/named.py": {
+        "size": 8618
+      },
+      "pyro/contrib/autoname/scoping.py": {
+        "size": 6482
+      },
+      "pyro/contrib/bnn/__init__.py": {
+        "size": 177
+      },
+      "pyro/contrib/bnn/hidden_layer.py": {
+        "size": 5384
+      },
+      "pyro/contrib/bnn/utils.py": {
+        "size": 490
+      },
+      "pyro/contrib/cevae/__init__.py": {
+        "size": 23079
+      },
+      "pyro/contrib/conjugate/__init__.py": {
+        "size": 0
+      },
+      "pyro/contrib/conjugate/infer.py": {
+        "size": 9073
+      },
+      "pyro/contrib/easyguide/__init__.py": {
+        "size": 206
+      },
+      "pyro/contrib/easyguide/easyguide.py": {
+        "size": 12912
+      },
+      "pyro/contrib/epidemiology/__init__.py": {
+        "size": 406
+      },
+      "pyro/contrib/epidemiology/compartmental.py": {
+        "size": 49835
+      },
+      "pyro/contrib/epidemiology/distributions.py": {
+        "size": 13525
+      },
+      "pyro/contrib/epidemiology/models.py": {
+        "size": 51215
+      },
+      "pyro/contrib/epidemiology/util.py": {
+        "size": 10993
+      },
+      "pyro/contrib/examples/__init__.py": {
+        "size": 0
+      },
+      "pyro/contrib/examples/bart.py": {
+        "size": 6760
+      },
+      "pyro/contrib/examples/finance.py": {
+        "size": 694
+      },
+      "pyro/contrib/examples/multi_mnist.py": {
+        "size": 2920
+      },
+      "pyro/contrib/examples/nextstrain.py": {
+        "size": 1547
+      },
+      "pyro/contrib/examples/polyphonic_data_loader.py": {
+        "size": 6875
+      },
+      "pyro/contrib/examples/scanvi_data.py": {
+        "size": 7442
+      },
+      "pyro/contrib/examples/util.py": {
+        "size": 1556
+      },
+      "pyro/contrib/forecast/__init__.py": {
+        "size": 360
+      },
+      "pyro/contrib/forecast/evaluate.py": {
+        "size": 8823
+      },
+      "pyro/contrib/forecast/forecaster.py": {
+        "size": 23353
+      },
+      "pyro/contrib/forecast/util.py": {
+        "size": 16054
+      },
+      "pyro/contrib/funsor/__init__.py": {
+        "size": 1223
+      },
+      "pyro/contrib/funsor/handlers/__init__.py": {
+        "size": 1408
+      },
+      "pyro/contrib/funsor/handlers/enum_messenger.py": {
+        "size": 9727
+      },
+      "pyro/contrib/funsor/handlers/named_messenger.py": {
+        "size": 7623
+      },
+      "pyro/contrib/funsor/handlers/plate_messenger.py": {
+        "size": 15224
+      },
+      "pyro/contrib/funsor/handlers/primitives.py": {
+        "size": 1076
+      },
+      "pyro/contrib/funsor/handlers/replay_messenger.py": {
+        "size": 1806
+      },
+      "pyro/contrib/funsor/handlers/runtime.py": {
+        "size": 8558
+      },
+      "pyro/contrib/funsor/handlers/trace_messenger.py": {
+        "size": 3632
+      },
+      "pyro/contrib/funsor/infer/__init__.py": {
+        "size": 514
+      },
+      "pyro/contrib/funsor/infer/discrete.py": {
+        "size": 2921
+      },
+      "pyro/contrib/funsor/infer/elbo.py": {
+        "size": 1652
+      },
+      "pyro/contrib/funsor/infer/trace_elbo.py": {
+        "size": 1724
+      },
+      "pyro/contrib/funsor/infer/traceenum_elbo.py": {
+        "size": 12731
+      },
+      "pyro/contrib/funsor/infer/tracetmc_elbo.py": {
+        "size": 1916
+      },
+      "pyro/contrib/gp/__init__.py": {
+        "size": 340
+      },
+      "pyro/contrib/gp/kernels/__init__.py": {
+        "size": 1533
+      },
+      "pyro/contrib/gp/kernels/brownian.py": {
+        "size": 1577
+      },
+      "pyro/contrib/gp/kernels/coregionalize.py": {
+        "size": 3699
+      },
+      "pyro/contrib/gp/kernels/dot_product.py": {
+        "size": 2646
+      },
+      "pyro/contrib/gp/kernels/isotropic.py": {
+        "size": 5740
+      },
+      "pyro/contrib/gp/kernels/kernel.py": {
+        "size": 8446
+      },
+      "pyro/contrib/gp/kernels/periodic.py": {
+        "size": 2424
+      },
+      "pyro/contrib/gp/kernels/static.py": {
+        "size": 1536
+      },
+      "pyro/contrib/gp/likelihoods/__init__.py": {
+        "size": 897
+      },
+      "pyro/contrib/gp/likelihoods/binary.py": {
+        "size": 1988
+      },
+      "pyro/contrib/gp/likelihoods/gaussian.py": {
+        "size": 1677
+      },
+      "pyro/contrib/gp/likelihoods/likelihood.py": {
+        "size": 875
+      },
+      "pyro/contrib/gp/likelihoods/multi_class.py": {
+        "size": 2859
+      },
+      "pyro/contrib/gp/likelihoods/poisson.py": {
+        "size": 1884
+      },
+      "pyro/contrib/gp/models/__init__.py": {
+        "size": 546
+      },
+      "pyro/contrib/gp/models/gplvm.py": {
+        "size": 3897
+      },
+      "pyro/contrib/gp/models/gpr.py": {
+        "size": 8932
+      },
+      "pyro/contrib/gp/models/model.py": {
+        "size": 9145
+      },
+      "pyro/contrib/gp/models/sgpr.py": {
+        "size": 11001
+      },
+      "pyro/contrib/gp/models/vgp.py": {
+        "size": 6859
+      },
+      "pyro/contrib/gp/models/vsgp.py": {
+        "size": 8417
+      },
+      "pyro/contrib/gp/parameterized.py": {
+        "size": 8052
+      },
+      "pyro/contrib/gp/util.py": {
+        "size": 7152
+      },
+      "pyro/contrib/minipyro.py": {
+        "size": 15591
+      },
+      "pyro/contrib/mue/__init__.py": {
+        "size": 0
+      },
+      "pyro/contrib/mue/dataloaders.py": {
+        "size": 6063
+      },
+      "pyro/contrib/mue/missingdatahmm.py": {
+        "size": 13358
+      },
+      "pyro/contrib/mue/models.py": {
+        "size": 34329
+      },
+      "pyro/contrib/mue/statearrangers.py": {
+        "size": 9017
+      },
+      "pyro/contrib/oed/__init__.py": {
+        "size": 3786
+      },
+      "pyro/contrib/oed/eig.py": {
+        "size": 44795
+      },
+      "pyro/contrib/oed/glmm/__init__.py": {
+        "size": 1551
+      },
+      "pyro/contrib/oed/glmm/glmm.py": {
+        "size": 16376
+      },
+      "pyro/contrib/oed/glmm/guides.py": {
+        "size": 11368
+      },
+      "pyro/contrib/oed/search.py": {
+        "size": 867
+      },
+      "pyro/contrib/oed/util.py": {
+        "size": 1230
+      },
+      "pyro/contrib/randomvariable/__init__.py": {
+        "size": 193
+      },
+      "pyro/contrib/randomvariable/random_variable.py": {
+        "size": 5424
+      },
+      "pyro/contrib/timeseries/__init__.py": {
+        "size": 711
+      },
+      "pyro/contrib/timeseries/base.py": {
+        "size": 2211
+      },
+      "pyro/contrib/timeseries/gp.py": {
+        "size": 23486
+      },
+      "pyro/contrib/timeseries/lgssm.py": {
+        "size": 6439
+      },
+      "pyro/contrib/timeseries/lgssmgp.py": {
+        "size": 11260
+      },
+      "pyro/contrib/tracking/__init__.py": {
+        "size": 167
+      },
+      "pyro/contrib/tracking/assignment.py": {
+        "size": 19243
+      },
+      "pyro/contrib/tracking/distributions.py": {
+        "size": 3577
+      },
+      "pyro/contrib/tracking/dynamic_models.py": {
+        "size": 16560
+      },
+      "pyro/contrib/tracking/extended_kalman_filter.py": {
+        "size": 7796
+      },
+      "pyro/contrib/tracking/hashing.py": {
+        "size": 7266
+      },
+      "pyro/contrib/tracking/measurements.py": {
+        "size": 4544
+      },
+      "pyro/contrib/util.py": {
+        "size": 2800
+      },
+      "pyro/contrib/zuko.py": {
+        "size": 2230
+      },
+      "pyro/distributions/__init__.py": {
+        "size": 7555
+      },
+      "pyro/distributions/affine_beta.py": {
+        "size": 3985
+      },
+      "pyro/distributions/asymmetriclaplace.py": {
+        "size": 7367
+      },
+      "pyro/distributions/avf_mvn.py": {
+        "size": 4062
+      },
+      "pyro/distributions/coalescent.py": {
+        "size": 20914
+      },
+      "pyro/distributions/conditional.py": {
+        "size": 4895
+      },
+      "pyro/distributions/conjugate.py": {
+        "size": 10898
+      },
+      "pyro/distributions/constraints.py": {
+        "size": 5739
+      },
+      "pyro/distributions/delta.py": {
+        "size": 3066
+      },
+      "pyro/distributions/diag_normal_mixture.py": {
+        "size": 10433
+      },
+      "pyro/distributions/diag_normal_mixture_shared_cov.py": {
+        "size": 9139
+      },
+      "pyro/distributions/distribution.py": {
+        "size": 8481
+      },
+      "pyro/distributions/empirical.py": {
+        "size": 6765
+      },
+      "pyro/distributions/extended.py": {
+        "size": 1885
+      },
+      "pyro/distributions/folded.py": {
+        "size": 1290
+      },
+      "pyro/distributions/gaussian_scale_mixture.py": {
+        "size": 7983
+      },
+      "pyro/distributions/grouped_normal_normal.py": {
+        "size": 6572
+      },
+      "pyro/distributions/hmm.py": {
+        "size": 54251
+      },
+      "pyro/distributions/improper_uniform.py": {
+        "size": 2448
+      },
+      "pyro/distributions/inverse_gamma.py": {
+        "size": 1489
+      },
+      "pyro/distributions/kl.py": {
+        "size": 1661
+      },
+      "pyro/distributions/lkj.py": {
+        "size": 2534
+      },
+      "pyro/distributions/log_normal_negative_binomial.py": {
+        "size": 5720
+      },
+      "pyro/distributions/logistic.py": {
+        "size": 5473
+      },
+      "pyro/distributions/mixture.py": {
+        "size": 6150
+      },
+      "pyro/distributions/multivariate_studentt.py": {
+        "size": 5193
+      },
+      "pyro/distributions/nanmasked.py": {
+        "size": 3381
+      },
+      "pyro/distributions/omt_mvn.py": {
+        "size": 3176
+      },
+      "pyro/distributions/one_one_matching.py": {
+        "size": 6714
+      },
+      "pyro/distributions/one_two_matching.py": {
+        "size": 8193
+      },
+      "pyro/distributions/ordered_logistic.py": {
+        "size": 2874
+      },
+      "pyro/distributions/polya_gamma.py": {
+        "size": 3090
+      },
+      "pyro/distributions/projected_normal.py": {
+        "size": 7441
+      },
+      "pyro/distributions/rejector.py": {
+        "size": 2933
+      },
+      "pyro/distributions/relaxed_straight_through.py": {
+        "size": 3637
+      },
+      "pyro/distributions/score_parts.py": {
+        "size": 1282
+      },
+      "pyro/distributions/sine_bivariate_von_mises.py": {
+        "size": 11562
+      },
+      "pyro/distributions/sine_skewed.py": {
+        "size": 7372
+      },
+      "pyro/distributions/softlaplace.py": {
+        "size": 2471
+      },
+      "pyro/distributions/spanning_tree.py": {
+        "size": 22257
+      },
+      "pyro/distributions/stable.py": {
+        "size": 9997
+      },
+      "pyro/distributions/stable_log_prob.py": {
+        "size": 6573
+      },
+      "pyro/distributions/testing/__init__.py": {
+        "size": 215
+      },
+      "pyro/distributions/testing/fakes.py": {
+        "size": 421
+      },
+      "pyro/distributions/testing/gof.py": {
+        "size": 10761
+      },
+      "pyro/distributions/testing/naive_dirichlet.py": {
+        "size": 1598
+      },
+      "pyro/distributions/testing/rejection_exponential.py": {
+        "size": 1167
+      },
+      "pyro/distributions/testing/rejection_gamma.py": {
+        "size": 8842
+      },
+      "pyro/distributions/testing/special.py": {
+        "size": 3610
+      },
+      "pyro/distributions/torch.py": {
+        "size": 15147
+      },
+      "pyro/distributions/torch_distribution.py": {
+        "size": 20865
+      },
+      "pyro/distributions/torch_patch.py": {
+        "size": 3044
+      },
+      "pyro/distributions/torch_transform.py": {
+        "size": 1452
+      },
+      "pyro/distributions/transforms/__init__.py": {
+        "size": 8118
+      },
+      "pyro/distributions/transforms/affine_autoregressive.py": {
+        "size": 16342
+      },
+      "pyro/distributions/transforms/affine_coupling.py": {
+        "size": 15805
+      },
+      "pyro/distributions/transforms/basic.py": {
+        "size": 2073
+      },
+      "pyro/distributions/transforms/batchnorm.py": {
+        "size": 6147
+      },
+      "pyro/distributions/transforms/block_autoregressive.py": {
+        "size": 11703
+      },
+      "pyro/distributions/transforms/cholesky.py": {
+        "size": 2117
+      },
+      "pyro/distributions/transforms/discrete_cosine.py": {
+        "size": 3383
+      },
+      "pyro/distributions/transforms/generalized_channel_permute.py": {
+        "size": 11680
+      },
+      "pyro/distributions/transforms/haar.py": {
+        "size": 2883
+      },
+      "pyro/distributions/transforms/householder.py": {
+        "size": 10536
+      },
+      "pyro/distributions/transforms/lower_cholesky_affine.py": {
+        "size": 2479
+      },
+      "pyro/distributions/transforms/matrix_exponential.py": {
+        "size": 13725
+      },
+      "pyro/distributions/transforms/neural_autoregressive.py": {
+        "size": 9794
+      },
+      "pyro/distributions/transforms/normalize.py": {
+        "size": 1072
+      },
+      "pyro/distributions/transforms/ordered.py": {
+        "size": 921
+      },
+      "pyro/distributions/transforms/permute.py": {
+        "size": 5300
+      },
+      "pyro/distributions/transforms/planar.py": {
+        "size": 8881
+      },
+      "pyro/distributions/transforms/polynomial.py": {
+        "size": 7076
+      },
+      "pyro/distributions/transforms/power.py": {
+        "size": 2139
+      },
+      "pyro/distributions/transforms/radial.py": {
+        "size": 8711
+      },
+      "pyro/distributions/transforms/simplex_to_ordered.py": {
+        "size": 2527
+      },
+      "pyro/distributions/transforms/softplus.py": {
+        "size": 1665
+      },
+      "pyro/distributions/transforms/spline.py": {
+        "size": 24458
+      },
+      "pyro/distributions/transforms/spline_autoregressive.py": {
+        "size": 11370
+      },
+      "pyro/distributions/transforms/spline_coupling.py": {
+        "size": 6917
+      },
+      "pyro/distributions/transforms/sylvester.py": {
+        "size": 6318
+      },
+      "pyro/distributions/transforms/unit_cholesky.py": {
+        "size": 809
+      },
+      "pyro/distributions/transforms/utils.py": {
+        "size": 282
+      },
+      "pyro/distributions/unit.py": {
+        "size": 1886
+      },
+      "pyro/distributions/util.py": {
+        "size": 11913
+      },
+      "pyro/distributions/von_mises_3d.py": {
+        "size": 2694
+      },
+      "pyro/distributions/zero_inflated.py": {
+        "size": 6893
+      },
+      "pyro/generic.py": {
+        "size": 266
+      },
+      "pyro/infer/__init__.py": {
+        "size": 2195
+      },
+      "pyro/infer/abstract_infer.py": {
+        "size": 15986
+      },
+      "pyro/infer/autoguide/__init__.py": {
+        "size": 1618
+      },
+      "pyro/infer/autoguide/effect.py": {
+        "size": 19652
+      },
+      "pyro/infer/autoguide/gaussian.py": {
+        "size": 26015
+      },
+      "pyro/infer/autoguide/guides.py": {
+        "size": 49204
+      },
+      "pyro/infer/autoguide/initialization.py": {
+        "size": 8270
+      },
+      "pyro/infer/autoguide/structured.py": {
+        "size": 16327
+      },
+      "pyro/infer/autoguide/utils.py": {
+        "size": 3089
+      },
+      "pyro/infer/csis.py": {
+        "size": 7251
+      },
+      "pyro/infer/discrete.py": {
+        "size": 11547
+      },
+      "pyro/infer/elbo.py": {
+        "size": 9535
+      },
+      "pyro/infer/energy_distance.py": {
+        "size": 10044
+      },
+      "pyro/infer/enum.py": {
+        "size": 8063
+      },
+      "pyro/infer/importance.py": {
+        "size": 9786
+      },
+      "pyro/infer/inspect.py": {
+        "size": 23989
+      },
+      "pyro/infer/mcmc/__init__.py": {
+        "size": 490
+      },
+      "pyro/infer/mcmc/adaptation.py": {
+        "size": 23000
+      },
+      "pyro/infer/mcmc/api.py": {
+        "size": 29278
+      },
+      "pyro/infer/mcmc/hmc.py": {
+        "size": 18825
+      },
+      "pyro/infer/mcmc/logger.py": {
+        "size": 9270
+      },
+      "pyro/infer/mcmc/mcmc_kernel.py": {
+        "size": 2347
+      },
+      "pyro/infer/mcmc/nuts.py": {
+        "size": 21893
+      },
+      "pyro/infer/mcmc/rwkernel.py": {
+        "size": 4827
+      },
+      "pyro/infer/mcmc/util.py": {
+        "size": 33005
+      },
+      "pyro/infer/predictive.py": {
+        "size": 25794
+      },
+      "pyro/infer/renyi_elbo.py": {
+        "size": 9418
+      },
+      "pyro/infer/reparam/__init__.py": {
+        "size": 1239
+      },
+      "pyro/infer/reparam/conjugate.py": {
+        "size": 4315
+      },
+      "pyro/infer/reparam/discrete_cosine.py": {
+        "size": 2076
+      },
+      "pyro/infer/reparam/haar.py": {
+        "size": 1594
+      },
+      "pyro/infer/reparam/hmm.py": {
+        "size": 6406
+      },
+      "pyro/infer/reparam/loc_scale.py": {
+        "size": 3812
+      },
+      "pyro/infer/reparam/neutra.py": {
+        "size": 5549
+      },
+      "pyro/infer/reparam/projected_normal.py": {
+        "size": 1697
+      },
+      "pyro/infer/reparam/reparam.py": {
+        "size": 2505
+      },
+      "pyro/infer/reparam/softmax.py": {
+        "size": 1869
+      },
+      "pyro/infer/reparam/split.py": {
+        "size": 2642
+      },
+      "pyro/infer/reparam/stable.py": {
+        "size": 10548
+      },
+      "pyro/infer/reparam/strategies.py": {
+        "size": 7528
+      },
+      "pyro/infer/reparam/structured.py": {
+        "size": 4312
+      },
+      "pyro/infer/reparam/studentt.py": {
+        "size": 1477
+      },
+      "pyro/infer/reparam/transform.py": {
+        "size": 1789
+      },
+      "pyro/infer/reparam/unit_jacobian.py": {
+        "size": 3913
+      },
+      "pyro/infer/resampler.py": {
+        "size": 5720
+      },
+      "pyro/infer/rws.py": {
+        "size": 11139
+      },
+      "pyro/infer/smcfilter.py": {
+        "size": 8713
+      },
+      "pyro/infer/svgd.py": {
+        "size": 12959
+      },
+      "pyro/infer/svi.py": {
+        "size": 5947
+      },
+      "pyro/infer/trace_elbo.py": {
+        "size": 10334
+      },
+      "pyro/infer/trace_mean_field_elbo.py": {
+        "size": 8302
+      },
+      "pyro/infer/trace_mmd.py": {
+        "size": 10876
+      },
+      "pyro/infer/trace_tail_adaptive_elbo.py": {
+        "size": 3367
+      },
+      "pyro/infer/traceenum_elbo.py": {
+        "size": 23531
+      },
+      "pyro/infer/tracegraph_elbo.py": {
+        "size": 16978
+      },
+      "pyro/infer/tracetmc_elbo.py": {
+        "size": 8714
+      },
+      "pyro/infer/util.py": {
+        "size": 13697
+      },
+      "pyro/logger.py": {
+        "size": 463
+      },
+      "pyro/nn/__init__.py": {
+        "size": 665
+      },
+      "pyro/nn/auto_reg_nn.py": {
+        "size": 14482
+      },
+      "pyro/nn/dense_nn.py": {
+        "size": 5553
+      },
+      "pyro/nn/module.py": {
+        "size": 40033
+      },
+      "pyro/ops/__init__.py": {
+        "size": 0
+      },
+      "pyro/ops/arrowhead.py": {
+        "size": 4306
+      },
+      "pyro/ops/contract.py": {
+        "size": 22550
+      },
+      "pyro/ops/dual_averaging.py": {
+        "size": 3402
+      },
+      "pyro/ops/einsum/__init__.py": {
+        "size": 1486
+      },
+      "pyro/ops/einsum/adjoint.py": {
+        "size": 4971
+      },
+      "pyro/ops/einsum/torch_log.py": {
+        "size": 2010
+      },
+      "pyro/ops/einsum/torch_map.py": {
+        "size": 1952
+      },
+      "pyro/ops/einsum/torch_marginal.py": {
+        "size": 2234
+      },
+      "pyro/ops/einsum/torch_sample.py": {
+        "size": 2736
+      },
+      "pyro/ops/einsum/util.py": {
+        "size": 1901
+      },
+      "pyro/ops/gamma_gaussian.py": {
+        "size": 17941
+      },
+      "pyro/ops/gaussian.py": {
+        "size": 28119
+      },
+      "pyro/ops/hessian.py": {
+        "size": 673
+      },
+      "pyro/ops/indexing.py": {
+        "size": 7458
+      },
+      "pyro/ops/integrator.py": {
+        "size": 5119
+      },
+      "pyro/ops/jit.py": {
+        "size": 5970
+      },
+      "pyro/ops/linalg.py": {
+        "size": 3577
+      },
+      "pyro/ops/newton.py": {
+        "size": 9520
+      },
+      "pyro/ops/packed.py": {
+        "size": 6350
+      },
+      "pyro/ops/provenance.py": {
+        "size": 5764
+      },
+      "pyro/ops/rings.py": {
+        "size": 11388
+      },
+      "pyro/ops/special.py": {
+        "size": 7046
+      },
+      "pyro/ops/ssm_gp.py": {
+        "size": 6648
+      },
+      "pyro/ops/stats.py": {
+        "size": 22039
+      },
+      "pyro/ops/streaming.py": {
+        "size": 8903
+      },
+      "pyro/ops/tensor_utils.py": {
+        "size": 15783
+      },
+      "pyro/ops/welford.py": {
+        "size": 3405
+      },
+      "pyro/optim/__init__.py": {
+        "size": 594
+      },
+      "pyro/optim/adagrad_rmsprop.py": {
+        "size": 3060
+      },
+      "pyro/optim/clipped_adam.py": {
+        "size": 3462
+      },
+      "pyro/optim/dct_adam.py": {
+        "size": 7634
+      },
+      "pyro/optim/horovod.py": {
+        "size": 1906
+      },
+      "pyro/optim/lr_scheduler.py": {
+        "size": 2351
+      },
+      "pyro/optim/multi.py": {
+        "size": 6391
+      },
+      "pyro/optim/optim.py": {
+        "size": 10610
+      },
+      "pyro/optim/pytorch_optimizers.py": {
+        "size": 1805
+      },
+      "pyro/params/__init__.py": {
+        "size": 317
+      },
+      "pyro/params/param_store.py": {
+        "size": 14507
+      },
+      "pyro/poutine/__init__.py": {
+        "size": 922
+      },
+      "pyro/poutine/block_messenger.py": {
+        "size": 5614
+      },
+      "pyro/poutine/broadcast_messenger.py": {
+        "size": 3722
+      },
+      "pyro/poutine/collapse_messenger.py": {
+        "size": 6395
+      },
+      "pyro/poutine/condition_messenger.py": {
+        "size": 2367
+      },
+      "pyro/poutine/do_messenger.py": {
+        "size": 3476
+      },
+      "pyro/poutine/enum_messenger.py": {
+        "size": 10711
+      },
+      "pyro/poutine/escape_messenger.py": {
+        "size": 1339
+      },
+      "pyro/poutine/guide.py": {
+        "size": 6163
+      },
+      "pyro/poutine/handlers.py": {
+        "size": 17835
+      },
+      "pyro/poutine/indep_messenger.py": {
+        "size": 4435
+      },
+      "pyro/poutine/infer_config_messenger.py": {
+        "size": 1997
+      },
+      "pyro/poutine/lift_messenger.py": {
+        "size": 5224
+      },
+      "pyro/poutine/markov_messenger.py": {
+        "size": 3658
+      },
+      "pyro/poutine/mask_messenger.py": {
+        "size": 1394
+      },
+      "pyro/poutine/messenger.py": {
+        "size": 9474
+      },
+      "pyro/poutine/plate_messenger.py": {
+        "size": 3167
+      },
+      "pyro/poutine/reentrant_messenger.py": {
+        "size": 1031
+      },
+      "pyro/poutine/reparam_messenger.py": {
+        "size": 6175
+      },
+      "pyro/poutine/replay_messenger.py": {
+        "size": 3310
+      },
+      "pyro/poutine/runtime.py": {
+        "size": 16650
+      },
+      "pyro/poutine/scale_messenger.py": {
+        "size": 1975
+      },
+      "pyro/poutine/seed_messenger.py": {
+        "size": 1308
+      },
+      "pyro/poutine/subsample_messenger.py": {
+        "size": 8509
+      },
+      "pyro/poutine/substitute_messenger.py": {
+        "size": 3295
+      },
+      "pyro/poutine/trace_messenger.py": {
+        "size": 7701
+      },
+      "pyro/poutine/trace_struct.py": {
+        "size": 22785
+      },
+      "pyro/poutine/uncondition_messenger.py": {
+        "size": 982
+      },
+      "pyro/poutine/util.py": {
+        "size": 4790
+      },
+      "pyro/primitives.py": {
+        "size": 24368
+      },
+      "pyro/settings.py": {
+        "size": 5019
+      },
+      "pyro/util.py": {
+        "size": 22051
+      },
+      "scripts/update_headers.py": {
+        "size": 2276
+      },
+      "scripts/update_version.py": {
+        "size": 1188
+      },
+      "setup.cfg": {
+        "size": 2382
+      },
+      "setup.py": {
+        "size": 4938
+      },
+      "tests/README.md": {
+        "size": 1962
+      },
+      "tests/__init__.py": {
+        "size": 282
+      },
+      "tests/common.py": {
+        "size": 9455
+      },
+      "tests/conftest.py": {
+        "size": 3380
+      },
+      "tests/contrib/__init__.py": {
+        "size": 0
+      },
+      "tests/contrib/autoname/test_autoname.py": {
+        "size": 9663
+      },
+      "tests/contrib/autoname/test_named.py": {
+        "size": 3676
+      },
+      "tests/contrib/autoname/test_scoping.py": {
+        "size": 4628
+      },
+      "tests/contrib/bnn/test_hidden_layer.py": {
+        "size": 2100
+      },
+      "tests/contrib/cevae/test_cevae.py": {
+        "size": 2368
+      },
+      "tests/contrib/conftest.py": {
+        "size": 449
+      },
+      "tests/contrib/easyguide/test_easyguide.py": {
+        "size": 8598
+      },
+      "tests/contrib/epidemiology/__init__.py": {
+        "size": 0
+      },
+      "tests/contrib/epidemiology/test_distributions.py": {
+        "size": 9012
+      },
+      "tests/contrib/epidemiology/test_models.py": {
+        "size": 22313
+      },
+      "tests/contrib/epidemiology/test_quant.py": {
+        "size": 925
+      },
+      "tests/contrib/epidemiology/test_util.py": {
+        "size": 1492
+      },
+      "tests/contrib/forecast/__init__.py": {
+        "size": 0
+      },
+      "tests/contrib/forecast/test_evaluate.py": {
+        "size": 4477
+      },
+      "tests/contrib/forecast/test_forecaster.py": {
+        "size": 11591
+      },
+      "tests/contrib/forecast/test_util.py": {
+        "size": 5539
+      },
+      "tests/contrib/funsor/conftest.py": {
+        "size": 602
+      },
+      "tests/contrib/funsor/test_enum_funsor.py": {
+        "size": 71964
+      },
+      "tests/contrib/funsor/test_infer_discrete.py": {
+        "size": 16722
+      },
+      "tests/contrib/funsor/test_named_handlers.py": {
+        "size": 5329
+      },
+      "tests/contrib/funsor/test_pyroapi_funsor.py": {
+        "size": 488
+      },
+      "tests/contrib/funsor/test_tmc.py": {
+        "size": 7191
+      },
+      "tests/contrib/funsor/test_valid_models_enum.py": {
+        "size": 17970
+      },
+      "tests/contrib/funsor/test_valid_models_plate.py": {
+        "size": 7546
+      },
+      "tests/contrib/funsor/test_valid_models_sequential_plate.py": {
+        "size": 4418
+      },
+      "tests/contrib/funsor/test_vectorized_markov.py": {
+        "size": 30783
+      },
+      "tests/contrib/gp/__init__.py": {
+        "size": 0
+      },
+      "tests/contrib/gp/test_conditional.py": {
+        "size": 2750
+      },
+      "tests/contrib/gp/test_kernels.py": {
+        "size": 4744
+      },
+      "tests/contrib/gp/test_likelihoods.py": {
+        "size": 4731
+      },
+      "tests/contrib/gp/test_models.py": {
+        "size": 15042
+      },
+      "tests/contrib/gp/test_parameterized.py": {
+        "size": 4723
+      },
+      "tests/contrib/mue/test_dataloaders.py": {
+        "size": 3607
+      },
+      "tests/contrib/mue/test_missingdatahmm.py": {
+        "size": 18642
+      },
+      "tests/contrib/mue/test_models.py": {
+        "size": 3213
+      },
+      "tests/contrib/mue/test_statearrangers.py": {
+        "size": 10625
+      },
+      "tests/contrib/oed/test_ewma.py": {
+        "size": 1464
+      },
+      "tests/contrib/oed/test_finite_spaces_eig.py": {
+        "size": 7851
+      },
+      "tests/contrib/oed/test_glmm.py": {
+        "size": 5729
+      },
+      "tests/contrib/oed/test_linear_models_eig.py": {
+        "size": 9466
+      },
+      "tests/contrib/oed/test_xexpx.py": {
+        "size": 470
+      },
+      "tests/contrib/randomvariable/test_random_variable.py": {
+        "size": 1893
+      },
+      "tests/contrib/test_hessian.py": {
+        "size": 853
+      },
+      "tests/contrib/test_minipyro.py": {
+        "size": 8701
+      },
+      "tests/contrib/test_util.py": {
+        "size": 3036
+      },
+      "tests/contrib/test_zuko.py": {
+        "size": 1671
+      },
+      "tests/contrib/timeseries/test_gp.py": {
+        "size": 5986
+      },
+      "tests/contrib/timeseries/test_lgssm.py": {
+        "size": 3905
+      },
+      "tests/contrib/tracking/__init__.py": {
+        "size": 0
+      },
+      "tests/contrib/tracking/test_assignment.py": {
+        "size": 14126
+      },
+      "tests/contrib/tracking/test_distributions.py": {
+        "size": 890
+      },
+      "tests/contrib/tracking/test_dynamic_models.py": {
+        "size": 5020
+      },
+      "tests/contrib/tracking/test_ekf.py": {
+        "size": 2792
+      },
+      "tests/contrib/tracking/test_em.py": {
+        "size": 8636
+      },
+      "tests/contrib/tracking/test_hashing.py": {
+        "size": 4601
+      },
+      "tests/contrib/tracking/test_measurements.py": {
+        "size": 934
+      },
+      "tests/distributions/__init__.py": {
+        "size": 0
+      },
+      "tests/distributions/conftest.py": {
+        "size": 35808
+      },
+      "tests/distributions/dist_fixture.py": {
+        "size": 6500
+      },
+      "tests/distributions/test_binomial.py": {
+        "size": 2014
+      },
+      "tests/distributions/test_categorical.py": {
+        "size": 3591
+      },
+      "tests/distributions/test_coalescent.py": {
+        "size": 11355
+      },
+      "tests/distributions/test_conjugate.py": {
+        "size": 4358
+      },
+      "tests/distributions/test_conjugate_update.py": {
+        "size": 2061
+      },
+      "tests/distributions/test_constraints.py": {
+        "size": 1082
+      },
+      "tests/distributions/test_cuda.py": {
+        "size": 3189
+      },
+      "tests/distributions/test_delta.py": {
+        "size": 2862
+      },
+      "tests/distributions/test_distributions.py": {
+        "size": 16165
+      },
+      "tests/distributions/test_empirical.py": {
+        "size": 7002
+      },
+      "tests/distributions/test_extended.py": {
+        "size": 3462
+      },
+      "tests/distributions/test_gaussian_mixtures.py": {
+        "size": 8768
+      },
+      "tests/distributions/test_grouped_normal_normal.py": {
+        "size": 1565
+      },
+      "tests/distributions/test_haar.py": {
+        "size": 399
+      },
+      "tests/distributions/test_hmm.py": {
+        "size": 36873
+      },
+      "tests/distributions/test_ig.py": {
+        "size": 1192
+      },
+      "tests/distributions/test_improper_uniform.py": {
+        "size": 966
+      },
+      "tests/distributions/test_independent.py": {
+        "size": 5135
+      },
+      "tests/distributions/test_kl.py": {
+        "size": 3702
+      },
+      "tests/distributions/test_lkj.py": {
+        "size": 4821
+      },
+      "tests/distributions/test_log_normal_negative_binomial.py": {
+        "size": 1588
+      },
+      "tests/distributions/test_lowrank_mvn.py": {
+        "size": 1451
+      },
+      "tests/distributions/test_mask.py": {
+        "size": 5258
+      },
+      "tests/distributions/test_mixture.py": {
+        "size": 5242
+      },
+      "tests/distributions/test_mvn.py": {
+        "size": 1425
+      },
+      "tests/distributions/test_mvt.py": {
+        "size": 5652
+      },
+      "tests/distributions/test_nanmasked.py": {
+        "size": 3318
+      },
+      "tests/distributions/test_omt_mvn.py": {
+        "size": 4649
+      },
+      "tests/distributions/test_one_hot_categorical.py": {
+        "size": 3921
+      },
+      "tests/distributions/test_one_one_matching.py": {
+        "size": 5643
+      },
+      "tests/distributions/test_one_two_matching.py": {
+        "size": 9925
+      },
+      "tests/distributions/test_ordered_logistic.py": {
+        "size": 4745
+      },
+      "tests/distributions/test_pickle.py": {
+        "size": 3011
+      },
+      "tests/distributions/test_polya_gamma.py": {
+        "size": 904
+      },
+      "tests/distributions/test_projected_normal.py": {
+        "size": 826
+      },
+      "tests/distributions/test_rejector.py": {
+        "size": 6748
+      },
+      "tests/distributions/test_relaxed_straight_through.py": {
+        "size": 3847
+      },
+      "tests/distributions/test_reshape.py": {
+        "size": 5906
+      },
+      "tests/distributions/test_shapes.py": {
+        "size": 2509
+      },
+      "tests/distributions/test_sine_bivariate_von_mises.py": {
+        "size": 4343
+      },
+      "tests/distributions/test_sine_skewed.py": {
+        "size": 2967
+      },
+      "tests/distributions/test_spanning_tree.py": {
+        "size": 7701
+      },
+      "tests/distributions/test_stable.py": {
+        "size": 5262
+      },
+      "tests/distributions/test_stable_log_prob.py": {
+        "size": 5282
+      },
+      "tests/distributions/test_tensor_type.py": {
+        "size": 2032
+      },
+      "tests/distributions/test_torch_patch.py": {
+        "size": 1025
+      },
+      "tests/distributions/test_transforms.py": {
+        "size": 22417
+      },
+      "tests/distributions/test_unit.py": {
+        "size": 966
+      },
+      "tests/distributions/test_util.py": {
+        "size": 13215
+      },
+      "tests/distributions/test_von_mises.py": {
+        "size": 5467
+      },
+      "tests/distributions/test_zero_inflated.py": {
+        "size": 4929
+      },
+      "tests/distributions/testing/test_gof.py": {
+        "size": 612
+      },
+      "tests/distributions/testing/test_special.py": {
+        "size": 431
+      },
+      "tests/doctest_fixtures.py": {
+        "size": 1075
+      },
+      "tests/infer/__init__.py": {
+        "size": 0
+      },
+      "tests/infer/autoguide/__init__.py": {
+        "size": 0
+      },
+      "tests/infer/autoguide/conftest.py": {
+        "size": 442
+      },
+      "tests/infer/autoguide/test_gaussian.py": {
+        "size": 29698
+      },
+      "tests/infer/autoguide/test_inference.py": {
+        "size": 8100
+      },
+      "tests/infer/autoguide/test_mean_field_entropy.py": {
+        "size": 1064
+      },
+      "tests/infer/conftest.py": {
+        "size": 432
+      },
+      "tests/infer/mcmc/__init__.py": {
+        "size": 0
+      },
+      "tests/infer/mcmc/test_adaptation.py": {
+        "size": 2472
+      },
+      "tests/infer/mcmc/test_hmc.py": {
+        "size": 11825
+      },
+      "tests/infer/mcmc/test_mcmc_api.py": {
+        "size": 12719
+      },
+      "tests/infer/mcmc/test_mcmc_util.py": {
+        "size": 4106
+      },
+      "tests/infer/mcmc/test_nuts.py": {
+        "size": 20404
+      },
+      "tests/infer/mcmc/test_rwkernel.py": {
+        "size": 1308
+      },
+      "tests/infer/mcmc/test_valid_models.py": {
+        "size": 17096
+      },
+      "tests/infer/reparam/__init__.py": {
+        "size": 0
+      },
+      "tests/infer/reparam/test_conjugate.py": {
+        "size": 9154
+      },
+      "tests/infer/reparam/test_discrete_cosine.py": {
+        "size": 4030
+      },
+      "tests/infer/reparam/test_haar.py": {
+        "size": 6392
+      },
+      "tests/infer/reparam/test_hmm.py": {
+        "size": 12130
+      },
+      "tests/infer/reparam/test_loc_scale.py": {
+        "size": 4077
+      },
+      "tests/infer/reparam/test_neutra.py": {
+        "size": 3101
+      },
+      "tests/infer/reparam/test_projected_normal.py": {
+        "size": 2118
+      },
+      "tests/infer/reparam/test_softmax.py": {
+        "size": 2305
+      },
+      "tests/infer/reparam/test_split.py": {
+        "size": 3082
+      },
+      "tests/infer/reparam/test_stable.py": {
+        "size": 6419
+      },
+      "tests/infer/reparam/test_strategies.py": {
+        "size": 10119
+      },
+      "tests/infer/reparam/test_structured.py": {
+        "size": 2029
+      },
+      "tests/infer/reparam/test_studentt.py": {
+        "size": 3021
+      },
+      "tests/infer/reparam/test_transform.py": {
+        "size": 2709
+      },
+      "tests/infer/reparam/test_unit_jacobian.py": {
+        "size": 2597
+      },
+      "tests/infer/reparam/util.py": {
+        "size": 1193
+      },
+      "tests/infer/test_abstract_infer.py": {
+        "size": 2732
+      },
+      "tests/infer/test_autoguide.py": {
+        "size": 51678
+      },
+      "tests/infer/test_compute_downstream_costs.py": {
+        "size": 29962
+      },
+      "tests/infer/test_conjugate_gradients.py": {
+        "size": 1616
+      },
+      "tests/infer/test_csis.py": {
+        "size": 2468
+      },
+      "tests/infer/test_discrete.py": {
+        "size": 14353
+      },
+      "tests/infer/test_elbo_mapdata.py": {
+        "size": 4196
+      },
+      "tests/infer/test_enum.py": {
+        "size": 145511
+      },
+      "tests/infer/test_gradient.py": {
+        "size": 15384
+      },
+      "tests/infer/test_inference.py": {
+        "size": 36696
+      },
+      "tests/infer/test_initialization.py": {
+        "size": 1066
+      },
+      "tests/infer/test_inspect.py": {
+        "size": 16351
+      },
+      "tests/infer/test_jit.py": {
+        "size": 19363
+      },
+      "tests/infer/test_multi_sample_elbos.py": {
+        "size": 1638
+      },
+      "tests/infer/test_predictive.py": {
+        "size": 12494
+      },
+      "tests/infer/test_resampler.py": {
+        "size": 1321
+      },
+      "tests/infer/test_sampling.py": {
+        "size": 3094
+      },
+      "tests/infer/test_smcfilter.py": {
+        "size": 8196
+      },
+      "tests/infer/test_svgd.py": {
+        "size": 5251
+      },
+      "tests/infer/test_tmc.py": {
+        "size": 10256
+      },
+      "tests/infer/test_util.py": {
+        "size": 2397
+      },
+      "tests/infer/test_valid_models.py": {
+        "size": 93251
+      },
+      "tests/integration_tests/__init__.py": {
+        "size": 0
+      },
+      "tests/integration_tests/conftest.py": {
+        "size": 451
+      },
+      "tests/integration_tests/test_conjugate_gaussian_models.py": {
+        "size": 25477
+      },
+      "tests/integration_tests/test_tracegraph_elbo.py": {
+        "size": 25003
+      },
+      "tests/nn/__init__.py": {
+        "size": 0
+      },
+      "tests/nn/conftest.py": {
+        "size": 429
+      },
+      "tests/nn/test_autoregressive.py": {
+        "size": 5466
+      },
+      "tests/nn/test_module.py": {
+        "size": 35007
+      },
+      "tests/ops/__init__.py": {
+        "size": 0
+      },
+      "tests/ops/conftest.py": {
+        "size": 430
+      },
+      "tests/ops/einsum/conftest.py": {
+        "size": 437
+      },
+      "tests/ops/einsum/test_adjoint.py": {
+        "size": 4418
+      },
+      "tests/ops/einsum/test_torch_log.py": {
+        "size": 1562
+      },
+      "tests/ops/gamma_gaussian.py": {
+        "size": 1763
+      },
+      "tests/ops/gaussian.py": {
+        "size": 1675
+      },
+      "tests/ops/test_arrowhead.py": {
+        "size": 1733
+      },
+      "tests/ops/test_contract.py": {
+        "size": 26142
+      },
+      "tests/ops/test_gamma_gaussian.py": {
+        "size": 12357
+      },
+      "tests/ops/test_gaussian.py": {
+        "size": 23038
+      },
+      "tests/ops/test_indexing.py": {
+        "size": 6117
+      },
+      "tests/ops/test_integrator.py": {
+        "size": 4558
+      },
+      "tests/ops/test_jit.py": {
+        "size": 1077
+      },
+      "tests/ops/test_linalg.py": {
+        "size": 2352
+      },
+      "tests/ops/test_newton.py": {
+        "size": 5028
+      },
+      "tests/ops/test_packed.py": {
+        "size": 2259
+      },
+      "tests/ops/test_provenance.py": {
+        "size": 1855
+      },
+      "tests/ops/test_special.py": {
+        "size": 2760
+      },
+      "tests/ops/test_ssm_gp.py": {
+        "size": 1153
+      },
+      "tests/ops/test_stats.py": {
+        "size": 10996
+      },
+      "tests/ops/test_streaming.py": {
+        "size": 3172
+      },
+      "tests/ops/test_tensor_utils.py": {
+        "size": 6592
+      },
+      "tests/ops/test_welford.py": {
+        "size": 3133
+      },
+      "tests/optim/__init__.py": {
+        "size": 0
+      },
+      "tests/optim/conftest.py": {
+        "size": 432
+      },
+      "tests/optim/test_multi.py": {
+        "size": 2637
+      },
+      "tests/optim/test_optim.py": {
+        "size": 14654
+      },
+      "tests/params/__init__.py": {
+        "size": 0
+      },
+      "tests/params/conftest.py": {
+        "size": 433
+      },
+      "tests/params/test_module.py": {
+        "size": 2567
+      },
+      "tests/params/test_param.py": {
+        "size": 8707
+      },
+      "tests/perf/__init__.py": {
+        "size": 0
+      },
+      "tests/perf/conftest.py": {
+        "size": 430
+      },
+      "tests/perf/test_benchmark.py": {
+        "size": 6793
+      },
+      "tests/poutine/__init__.py": {
+        "size": 0
+      },
+      "tests/poutine/conftest.py": {
+        "size": 434
+      },
+      "tests/poutine/test_counterfactual.py": {
+        "size": 3829
+      },
+      "tests/poutine/test_mapdata.py": {
+        "size": 5407
+      },
+      "tests/poutine/test_nesting.py": {
+        "size": 1068
+      },
+      "tests/poutine/test_poutines.py": {
+        "size": 36326
+      },
+      "tests/poutine/test_properties.py": {
+        "size": 3965
+      },
+      "tests/poutine/test_runtime.py": {
+        "size": 1113
+      },
+      "tests/poutine/test_trace_struct.py": {
+        "size": 1742
+      },
+      "tests/pyroapi/conftest.py": {
+        "size": 234
+      },
+      "tests/pyroapi/test_pyroapi.py": {
+        "size": 338
+      },
+      "tests/test_examples.py": {
+        "size": 22086
+      },
+      "tests/test_generic.py": {
+        "size": 3354
+      },
+      "tests/test_primitives.py": {
+        "size": 1459
+      },
+      "tests/test_settings.py": {
+        "size": 1420
+      },
+      "tests/test_util.py": {
+        "size": 3150
+      },
+      "tutorial/README.md": {
+        "size": 51
+      },
+      "tutorial/source/cleannb.py": {
+        "size": 1002
+      },
+      "tutorial/source/conf.py": {
+        "size": 5615
+      },
+      "tutorial/source/search_inference.py": {
+        "size": 7589
+      }
+    },
+    "processed_by": "zip_fallback",
+    "success": true
+  },
+  "structure": {
+    "packages": [
+      "source.examples",
+      "source.examples.contrib",
+      "source.examples.cvae",
+      "source.examples.mixed_hmm",
+      "source.examples.scanvi",
+      "source.profiler",
+      "source.pyro",
+      "source.pyro.contrib",
+      "source.pyro.distributions",
+      "source.pyro.infer",
+      "source.pyro.nn",
+      "source.pyro.ops",
+      "source.pyro.optim",
+      "source.pyro.params",
+      "source.pyro.poutine",
+      "source.tests",
+      "source.tests.contrib",
+      "source.tests.distributions",
+      "source.tests.infer",
+      "source.tests.integration_tests",
+      "source.tests.nn",
+      "source.tests.ops",
+      "source.tests.optim",
+      "source.tests.params",
+      "source.tests.perf",
+      "source.tests.poutine"
+    ]
+  },
+  "dependencies": {
+    "has_environment_yml": false,
+    "has_requirements_txt": false,
+    "pyproject": true,
+    "setup_cfg": true,
+    "setup_py": true
+  },
+  "entry_points": {
+    "imports": [],
+    "cli": [],
+    "modules": []
+  },
+  "llm_analysis": {
+    "core_modules": [
+      {
+        "package": "source.pyro",
+        "module": "pyro",
+        "functions": [
+          "sample",
+          "param",
+          "plate",
+          "factor",
+          "deterministic"
+        ],
+        "classes": [],
+        "description": "Core primitives for probabilistic programming in Pyro."
+      },
+      {
+        "package": "source.pyro.poutine",
+        "module": "poutine",
+        "functions": [
+          "trace",
+          "replay",
+          "condition",
+          "block",
+          "scale",
+          "enum"
+        ],
+        "classes": [
+          "Messenger"
+        ],
+        "description": "Effect handler system for intercepting and transforming probabilistic primitives."
+      },
+      {
+        "package": "source.pyro.infer",
+        "module": "infer",
+        "functions": [
+          "SVI",
+          "MCMC",
+          "Importance"
+        ],
+        "classes": [],
+        "description": "Inference algorithms including SVI and MCMC methods."
+      },
+      {
+        "package": "source.pyro.distributions",
+        "module": "distributions",
+        "functions": [],
+        "classes": [
+          "Delta",
+          "HiddenMarkovModel",
+          "Empirical"
+        ],
+        "description": "Extensions to PyTorch's distribution library with additional distributions and functionality."
+      }
+    ],
+    "cli_commands": [],
+    "import_strategy": {
+      "primary": "import",
+      "fallback": "blackbox",
+      "confidence": 0.9
+    },
+    "dependencies": {
+      "required": [
+        "torch",
+        "numpy"
+      ],
+      "optional": [
+        "matplotlib",
+        "scipy"
+      ]
+    },
+    "risk_assessment": {
+      "import_feasibility": 0.9,
+      "intrusiveness_risk": "low",
+      "complexity": "medium"
+    }
+  },
+  "deepwiki_analysis": {
+    "repo_url": "https://github.com/pyro-ppl/pyro",
+    "repo_name": "pyro",
+    "content": "pyro-ppl/pyro\nCore Primitives and Effect Handlers\nEffect Handlers\nTrace System\nDistributions\nDistribution Classes\nTransforms and Constraints\nSpecial Distributions\nDistribution Utilities\nInference Algorithms\nStochastic Variational Inference\nELBO Variants\nMCMC Methods\nImportance Sampling and Predictive\nModels and Guides\nThe Model/Guide Pattern\nParameter Management\nParameter Store\nExamples and Applications\nTutorial System\nBayesian Statistical Models\nDeep Generative Models\nSpecialized Applications\nDevelopment Infrastructure\nSetup and Installation\nTesting Framework\nSpecial Operations and Utilities\ndocs/requirements.txt\ndocs/source/conf.py\npyproject.toml\npyro/__init__.py\npyro/distributions/constraints.py\npyro/poutine/__init__.py\npyro/util.py\nThis page provides a high-level introduction to Pyro's architecture, design philosophy, and core components. It serves as an entry point for understanding how the system is structured and how its parts interact.\nFor detailed information about specific subsystems, see:\nCore primitives and effect handlers:Section 2\nDistribution system:Section 3\nInference algorithms:Section 4\nModel and guide patterns:Section 5\nParameter management:Section 6\nWhat is Pyro?\nPyro is a universal probabilistic programming language (PPL) built on PyTorch. It enables users to write generative models as Python programs and perform inference over them using a variety of algorithms including variational inference (SVI) and Markov chain Monte Carlo (MCMC).\nPackage Identity:\nName:pyro-pplsetup.py86\nDescription:\"A Python library for probabilistic modeling and inference\"setup.py88\nVersion:1.9.1+pyro/__init__.py31\nPython Requirements:>=3.8setup.py147\nCore Dependencies:\nSources:setup.py102-111pyro/__init__.py1-67\nDesign Philosophy\nPyro is built on four core principles that guide its architecture:\nUniversalAny computable distributionScalableMinimal overhead vs hand-written codeMinimalSmall core, powerful abstractionsFlexibleAutomation + manual controlPyro Core System\nUniversalAny computable distribution\nScalableMinimal overhead vs hand-written code\nMinimalSmall core, powerful abstractions\nFlexibleAutomation + manual control\nPyro Core System\nPyro can represent any computable probability distribution through its flexible programming model. Models are ordinary Python functions that can include loops, recursion, and arbitrary control flow.\nPyro is designed to scale to large datasets with minimal overhead. It builds directly on PyTorch's tensor operations and automatic differentiation, avoiding unnecessary abstractions.\nThe core system is implemented with a small set of composable abstractions. The effect handler system (poutine) provides the foundation for model manipulation without requiring extensive framework code.\nPyro provides high-level automation (e.g., automatic variational guides) while allowing experts to customize inference at any level through effect handlers and custom implementations.\nSources:README.md24-29\nSystem Architecture Overview\nPyro's architecture consists of several layered subsystems:\nParameter LayerDistribution LayerInference LayerEffect Handler Layer (Poutine)Primitive LayerUser LayerUser Model CodePython functions withpyro.sample, pyro.paramPyroModuleNeural network integrationExamples & TutorialsVAE, GP, HMM, etc.Primitive Operationspyro.samplepyro.parampyro.platepyro.factorMessage Runtime_PYRO_STACKapply_stack()Effect Handlerstrace, replaycondition, blockscale, enumSVIStochastic Variational InferenceMCMCHMC, NUTSAutoGuideAutomatic guide generationELBO VariantsTrace_ELBOTraceEnum_ELBOPyTorch Distributionstorch.distributions.*Pyro DistributionsDelta, HMM, etc.Constraints & TransformsParamStoreDictGlobal parameter storagePyroOptimWrapped PyTorch optimizers\nParameter Layer\nDistribution Layer\nInference Layer\nEffect Handler Layer (Poutine)\nPrimitive Layer\nUser Model CodePython functions withpyro.sample, pyro.param\nPyroModuleNeural network integration\nExamples & TutorialsVAE, GP, HMM, etc.\nPrimitive Operationspyro.samplepyro.parampyro.platepyro.factor\nMessage Runtime_PYRO_STACKapply_stack()\nEffect Handlerstrace, replaycondition, blockscale, enum\nSVIStochastic Variational Inference\nMCMCHMC, NUTS\nAutoGuideAutomatic guide generation\nELBO VariantsTrace_ELBOTraceEnum_ELBO\nPyTorch Distributionstorch.distributions.*\nPyro DistributionsDelta, HMM, etc.\nConstraints & Transforms\nParamStoreDictGlobal parameter storage\nPyroOptimWrapped PyTorch optimizers\nSources:pyro/__init__.py1-67pyro/poutine/__init__.py1-59\nCore Components and Code Entities\nPrimitives Module\nThe top-level module exports the fundamental primitives that users interact with:\npyro.sample\npyro.primitives\npyro.primitives\npyro.primitives\npyro.factor\npyro.primitives\npyro.deterministic\npyro.primitives\nAdditional utilities:\npyro.enable_validation()- Enable shape and distribution validationpyro/__init__.py12\npyro.enable_validation()\npyro.clear_param_store()- Reset global parameter storagepyro/__init__.py10\npyro.clear_param_store()\npyro.get_param_store()- Access the global parameter storepyro/__init__.py14\npyro.get_param_store()\npyro.set_rng_seed()- Set random number generator seedpyro/util.py37-45\npyro.set_rng_seed()\nSources:pyro/__init__.py8-26pyro/__init__.py39-66\nPoutine (Effect Handler) System\nPoutine is the effect handler system that intercepts and transforms probabilistic primitives:\nTrace Data StructureCore RuntimeHandler ExportstraceRecords executionreplayReplays recorded samplesconditionFixes observationsblockHides sites from handlersscaleScales log probabilitiesenumEnumerates discrete variablesreparamApplies reparameterization_PYRO_STACKGlobal message stackapply_stack()Message dispatcherMessengerBase handler classTraceExecution recordingMessageSite information dict\nTrace Data Structure\nCore Runtime\nHandler Exports\ntraceRecords execution\nreplayReplays recorded samples\nconditionFixes observations\nblockHides sites from handlers\nscaleScales log probabilities\nenumEnumerates discrete variables\nreparamApplies reparameterization\n_PYRO_STACKGlobal message stack\napply_stack()Message dispatcher\nMessengerBase handler class\nTraceExecution recording\nMessageSite information dict\nKey handler exports frompyro.poutine:\npyro.poutine\ntrace- Records model execution into a Trace objectpyro/poutine/__init__.py23\nreplay- Re-executes model with fixed sample valuespyro/poutine/__init__.py19\ncondition- Conditions model on observed datapyro/poutine/__init__.py8\nblock- Blocks primitives from outer handlerspyro/poutine/__init__.py5\nscale- Scales log probabilities (for subsampling)pyro/poutine/__init__.py20\nenum- Enumerates discrete latent variablespyro/poutine/__init__.py10\nSources:pyro/poutine/__init__.py1-59\nProgramming Model: From Code to Execution\nThis diagram shows how user code flows through Pyro's execution model:\n\"ParamStoreDict\"\"Distribution\"\"Effect Handlers\"\"_PYRO_STACK\"\"pyro.sample()\"\"User Code\"\"ParamStoreDict\"\"Distribution\"\"Effect Handlers\"\"_PYRO_STACK\"\"pyro.sample()\"\"User Code\"Message contains:name, fn, value, args, kwargsEach handler transforms messagePost-process (top-to-bottom)alt[If pyro.param() called]pyro.sample(\"x\", dist)Create messageProcess message (bottom-to-top)_process_message()Sample from distributionSample value_postprocess_message()Return processed messagevalueReturn samplepyro.param(\"theta\")Get/create parameterParameter tensorReturn parameter\nSources:pyro/__init__.py8-26pyro/poutine/__init__.py27-28\nParameter Management Architecture\nPyro maintains a global parameter store that synchronizes with PyTorch module parameters:\nOptimizationGlobal StorageUser Interfacepyro.param('theta', init)Get/create parameterPyroModule.x = PyroParam(...)Module-level parameterParamStoreDict_params: Dict[str, Tensor]_constraints: Dict[str, Constraint]pyro.get_param_store()Access singletonpyro.clear_param_store()Reset all parametersPyroOptimWraps torch.optimSVI.step()Optimization step\nOptimization\nGlobal Storage\nUser Interface\npyro.param('theta', init)Get/create parameter\nPyroModule.x = PyroParam(...)Module-level parameter\nParamStoreDict_params: Dict[str, Tensor]_constraints: Dict[str, Constraint]\npyro.get_param_store()Access singleton\npyro.clear_param_store()Reset all parameters\nPyroOptimWraps torch.optim\nSVI.step()Optimization step\nKey operations:\npyro.param(name, init_value, constraint=constraints.real)- Declare/retrieve parameter\npyro.param(name, init_value, constraint=constraints.real)\npyro.get_param_store()- Get the global ParamStoreDict instancepyro/__init__.py14\npyro.get_param_store()\npyro.clear_param_store()- Clear all parameterspyro/__init__.py10\npyro.clear_param_store()\nSources:pyro/__init__.py10-18pyro/__init__.py54\nDistribution System\nPyro extends PyTorch's distribution library with additional distributions and functionality:\nPyro-SpecificStandard DistributionsPyro ExtensionsPyTorch Basetorch.distributions.Distributiontorch.distributions.constraintsTorchDistributionMixinPyro compatibility layerpyro.distributions.constraintsAdditional constraintsNormal, Beta, Gamma, etc.Wrapped PyTorch distributionsDeltaDegenerate distributionHiddenMarkovModelDiscreteHMM, GaussianHMMEmpiricalSample-based distribution\nPyro-Specific\nStandard Distributions\nPyro Extensions\nPyTorch Base\ntorch.distributions.Distribution\ntorch.distributions.constraints\nTorchDistributionMixinPyro compatibility layer\npyro.distributions.constraintsAdditional constraints\nNormal, Beta, Gamma, etc.Wrapped PyTorch distributions\nDeltaDegenerate distribution\nHiddenMarkovModelDiscreteHMM, GaussianHMM\nEmpiricalSample-based distribution\nConstraint extensionsinpyro.distributions.constraints:\npyro.distributions.constraints\ninteger- Integer valuespyro/distributions/constraints.py152\nsphere- Points on Euclidean spherepyro/distributions/constraints.py155\ncorr_matrix- Correlation matricespyro/distributions/constraints.py151\ncorr_matrix\nordered_vector- Monotonically increasing vectorspyro/distributions/constraints.py153\nordered_vector\nAll PyTorch constraints viafrom torch.distributions.constraints import *pyro/distributions/constraints.py5\nfrom torch.distributions.constraints import *\nSources:pyro/distributions/constraints.py1-235\nInference Ecosystem\nPyro provides multiple inference algorithms that operate on the same model-guide abstraction:\npyro.infer.SVI\npyro.infer.MCMC\npyro.infer.MCMC\npyro.infer.Importance\nELBO variants for SVI:\nTrace_ELBO- Basic Monte Carlo ELBO estimator\nTraceEnum_ELBO- Enumerates discrete latent variables\nTraceEnum_ELBO\nTraceGraph_ELBO- Variance reduction via Rao-Blackwellization\nTraceGraph_ELBO\nRenyiELBO- Rényi divergence objective (α-divergence)\nAutomatic guides:",
+    "model": "gpt-4o-2024-08-06",
+    "source": "selenium",
+    "success": true
+  },
+  "deepwiki_options": {
+    "enabled": true,
+    "model": "gpt-4o-2024-08-06"
+  },
+  "risk": {
+    "import_feasibility": 0.9,
+    "intrusiveness_risk": "low",
+    "complexity": "medium"
+  }
+}

pyro/mcp_output/diff_report.md

@@ -0,0 +1,63 @@
+# Pyro Project Difference Report
+
+**Repository:** Pyro  
+**Project Type:** Python Library  
+**Main Features:** Basic Functionality  
+**Report Date:** February 7, 2026  
+**Intrusiveness:** None  
+**New Files:** 8  
+**Modified Files:** 0  
+**Workflow Status:** Success  
+**Test Status:** Failed  
+
+---
+
+## Project Overview
+
+The Pyro project is a Python library designed to provide basic functionality for probabilistic programming. As of the latest update, the project has introduced new files but has not modified any existing ones. The workflow has been executed successfully, but the test suite has failed, indicating potential issues with the new additions.
+
+## Difference Analysis
+
+### New Files
+
+The project has added 8 new files. These files likely introduce new features or components to the library. However, without modifications to existing files, it suggests that the new additions are standalone and do not alter the current functionality of the library.
+
+### Modified Files
+
+There are no modified files in this update. This indicates that the existing codebase remains unchanged, and the focus has been on expanding the library's capabilities through new files.
+
+### Workflow and Test Status
+
+- **Workflow Status:** The workflow has been successfully executed, indicating that the integration of new files did not disrupt the build or deployment processes.
+- **Test Status:** The test suite has failed, suggesting that the new files may contain errors or that they do not integrate well with the existing codebase.
+
+## Technical Analysis
+
+The introduction of new files without modifications to existing ones suggests an expansion of the library's capabilities. However, the failure in the test suite highlights potential issues:
+
+- **Integration Issues:** The new files may not be fully integrated with the existing codebase, leading to test failures.
+- **Incomplete Features:** The new features might be incomplete or not fully tested, resulting in errors during testing.
+- **Dependency Conflicts:** There could be conflicts with existing dependencies or assumptions within the codebase.
+
+## Recommendations and Improvements
+
+1. **Review New Files:** Conduct a thorough review of the new files to ensure they meet the project's coding standards and integrate seamlessly with the existing codebase.
+2. **Enhance Testing:** Update the test suite to cover the new features and ensure comprehensive testing of all functionalities.
+3. **Debugging:** Identify and resolve the specific issues causing test failures. This may involve debugging the new files or adjusting the test cases.
+4. **Documentation:** Update the project documentation to include details about the new features and any changes in usage or dependencies.
+
+## Deployment Information
+
+Given the successful workflow status, the new files have been integrated into the project without deployment issues. However, due to the test failures, it is advisable to hold off on deploying these changes to production until the issues are resolved.
+
+## Future Planning
+
+- **Stabilization:** Focus on stabilizing the current release by addressing test failures and ensuring all new features are fully functional.
+- **Feature Expansion:** Plan for future updates that expand on the new features, ensuring they are well-integrated and thoroughly tested.
+- **Community Feedback:** Engage with the user community to gather feedback on the new features and identify areas for improvement.
+
+## Conclusion
+
+The Pyro project has expanded its capabilities with the addition of new files. However, the test failures indicate that further work is needed to ensure these new features are stable and reliable. By addressing the identified issues and enhancing the testing framework, the project can continue to grow and provide valuable functionality to its users.
+
+---

pyro/mcp_output/mcp_plugin/adapter.py

@@ -0,0 +1,211 @@
+import os
+import sys
+
+# Path settings
+source_path = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), "source")
+sys.path.insert(0, source_path)
+
+# Import statements
+try:
+    from pyro.infer import SVI, MCMC, Importance
+    from pyro.infer.elbo import Trace_ELBO, TraceEnum_ELBO, TraceGraph_ELBO, RenyiELBO
+    from pyro.distributions import Delta, HiddenMarkovModel, Empirical
+    from pyro.poutine import trace, replay, condition, block, scale, enum
+    from pyro import sample, param, plate, factor, deterministic
+    from pyro.util import set_rng_seed, clear_param_store, get_param_store
+    import torch
+    import numpy as np
+except ImportError as e:
+    print(f"Import failed: {e}. Ensure all dependencies are installed.")
+    # Fallback mode
+    class Adapter:
+        def __init__(self):
+            self.mode = "blackbox"
+            print("Running in fallback mode. Limited functionality available.")
+
+        def status(self):
+            return {"status": "fallback", "message": "Limited functionality available due to import failure."}
+
+else:
+    # Adapter class definition
+    class Adapter:
+        """
+        Adapter class to interface with the Pyro library, providing methods to utilize its core functionalities.
+        """
+
+        def __init__(self):
+            self.mode = "import"
+
+        # -------------------- Inference Methods --------------------
+
+        def create_svi(self, model, guide, optim, loss=Trace_ELBO()):
+            """
+            Create an instance of the SVI class.
+
+            :param model: The model function.
+            :param guide: The guide function.
+            :param optim: The optimizer.
+            :param loss: The loss function, default is Trace_ELBO.
+            :return: Dictionary with status and SVI instance or error message.
+            """
+            try:
+                svi_instance = SVI(model, guide, optim, loss)
+                return {"status": "success", "svi_instance": svi_instance}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        def create_mcmc(self, kernel, num_samples, warmup_steps):
+            """
+            Create an instance of the MCMC class.
+
+            :param kernel: The MCMC kernel.
+            :param num_samples: Number of samples to draw.
+            :param warmup_steps: Number of warmup steps.
+            :return: Dictionary with status and MCMC instance or error message.
+            """
+            try:
+                mcmc_instance = MCMC(kernel, num_samples, warmup_steps)
+                return {"status": "success", "mcmc_instance": mcmc_instance}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        def create_importance(self, model, guide, num_samples):
+            """
+            Create an instance of the Importance class.
+
+            :param model: The model function.
+            :param guide: The guide function.
+            :param num_samples: Number of samples for importance sampling.
+            :return: Dictionary with status and Importance instance or error message.
+            """
+            try:
+                importance_instance = Importance(model, guide, num_samples)
+                return {"status": "success", "importance_instance": importance_instance}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        # -------------------- Distribution Methods --------------------
+
+        def create_delta(self, v, log_density=0.0):
+            """
+            Create an instance of the Delta distribution.
+
+            :param v: The value of the distribution.
+            :param log_density: The log density, default is 0.0.
+            :return: Dictionary with status and Delta instance or error message.
+            """
+            try:
+                delta_instance = Delta(v, log_density)
+                return {"status": "success", "delta_instance": delta_instance}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        def create_hidden_markov_model(self, initial_dist, transition_dist, observation_dist):
+            """
+            Create an instance of the HiddenMarkovModel.
+
+            :param initial_dist: Initial distribution.
+            :param transition_dist: Transition distribution.
+            :param observation_dist: Observation distribution.
+            :return: Dictionary with status and HiddenMarkovModel instance or error message.
+            """
+            try:
+                hmm_instance = HiddenMarkovModel(initial_dist, transition_dist, observation_dist)
+                return {"status": "success", "hmm_instance": hmm_instance}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        def create_empirical(self, samples):
+            """
+            Create an instance of the Empirical distribution.
+
+            :param samples: Samples for the empirical distribution.
+            :return: Dictionary with status and Empirical instance or error message.
+            """
+            try:
+                empirical_instance = Empirical(samples)
+                return {"status": "success", "empirical_instance": empirical_instance}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        # -------------------- Poutine Methods --------------------
+
+        def apply_trace(self, fn):
+            """
+            Apply the trace poutine to a function.
+
+            :param fn: The function to trace.
+            :return: Dictionary with status and traced function or error message.
+            """
+            try:
+                traced_fn = trace(fn)
+                return {"status": "success", "traced_fn": traced_fn}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        def apply_replay(self, fn, guide_trace):
+            """
+            Apply the replay poutine to a function.
+
+            :param fn: The function to replay.
+            :param guide_trace: The guide trace for replay.
+            :return: Dictionary with status and replayed function or error message.
+            """
+            try:
+                replayed_fn = replay(fn, guide_trace)
+                return {"status": "success", "replayed_fn": replayed_fn}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        def apply_condition(self, fn, data):
+            """
+            Apply the condition poutine to a function.
+
+            :param fn: The function to condition.
+            :param data: The observed data.
+            :return: Dictionary with status and conditioned function or error message.
+            """
+            try:
+                conditioned_fn = condition(fn, data)
+                return {"status": "success", "conditioned_fn": conditioned_fn}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        # -------------------- Utility Methods --------------------
+
+        def set_random_seed(self, seed):
+            """
+            Set the random number generator seed.
+
+            :param seed: The seed value.
+            :return: Dictionary with status.
+            """
+            try:
+                set_rng_seed(seed)
+                return {"status": "success"}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        def clear_parameters(self):
+            """
+            Clear all parameters from the parameter store.
+
+            :return: Dictionary with status.
+            """
+            try:
+                clear_param_store()
+                return {"status": "success"}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}
+
+        def get_parameters(self):
+            """
+            Get the global parameter store.
+
+            :return: Dictionary with status and parameter store or error message.
+            """
+            try:
+                param_store = get_param_store()
+                return {"status": "success", "param_store": param_store}
+            except Exception as e:
+                return {"status": "error", "message": str(e)}

pyro/mcp_output/mcp_plugin/main.py

@@ -0,0 +1,13 @@
+"""
+MCP Service Auto-Wrapper - Auto-generated
+"""
+from mcp_service import create_app
+
+def main():
+    """Main entry point"""
+    app = create_app()
+    return app
+
+if __name__ == "__main__":
+    app = main()
+    app.run()

pyro/mcp_output/mcp_plugin/mcp_service.py

@@ -0,0 +1,119 @@
+import os
+import sys
+
+# Add the local source directory to sys.path
+source_path = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), "source")
+if source_path not in sys.path:
+    sys.path.insert(0, source_path)
+
+from fastmcp import FastMCP
+
+# Import core modules from the local source directory
+from pyro.infer import SVI, MCMC
+from pyro.distributions import Delta, HiddenMarkovModel
+from pyro.poutine import trace, replay, condition
+
+# Create the FastMCP service application
+mcp = FastMCP("pyro_service")
+
+@mcp.tool(name="svi_tool", description="Perform Stochastic Variational Inference")
+def svi_tool(model: callable, guide: callable, data: dict) -> dict:
+    """
+    Perform Stochastic Variational Inference using the provided model and guide.
+
+    Parameters:
+    - model: The probabilistic model function.
+    - guide: The variational guide function.
+    - data: A dictionary containing observed data.
+
+    Returns:
+    - A dictionary with success, result, and error fields.
+    """
+    try:
+        svi = SVI(model, guide, ...)
+        result = svi.run(data)
+        return {"success": True, "result": result, "error": None}
+    except Exception as e:
+        return {"success": False, "result": None, "error": str(e)}
+
+@mcp.tool(name="mcmc_tool", description="Perform Markov Chain Monte Carlo")
+def mcmc_tool(model: callable, num_samples: int) -> dict:
+    """
+    Perform Markov Chain Monte Carlo sampling using the provided model.
+
+    Parameters:
+    - model: The probabilistic model function.
+    - num_samples: The number of samples to draw.
+
+    Returns:
+    - A dictionary with success, result, and error fields.
+    """
+    try:
+        mcmc = MCMC(model, num_samples=num_samples)
+        result = mcmc.run()
+        return {"success": True, "result": result, "error": None}
+    except Exception as e:
+        return {"success": False, "result": None, "error": str(e)}
+
+@mcp.tool(name="delta_distribution_tool", description="Create a Delta distribution")
+def delta_distribution_tool(value: float) -> dict:
+    """
+    Create a Delta distribution with the specified value.
+
+    Parameters:
+    - value: The value for the Delta distribution.
+
+    Returns:
+    - A dictionary with success, result, and error fields.
+    """
+    try:
+        delta_dist = Delta(value)
+        return {"success": True, "result": delta_dist, "error": None}
+    except Exception as e:
+        return {"success": False, "result": None, "error": str(e)}
+
+@mcp.tool(name="hmm_tool", description="Create a Hidden Markov Model")
+def hmm_tool(transitions: list, emissions: list, initial: list) -> dict:
+    """
+    Create a Hidden Markov Model with the specified parameters.
+
+    Parameters:
+    - transitions: Transition probabilities.
+    - emissions: Emission probabilities.
+    - initial: Initial state probabilities.
+
+    Returns:
+    - A dictionary with success, result, and error fields.
+    """
+    try:
+        hmm = HiddenMarkovModel(transitions, emissions, initial)
+        return {"success": True, "result": hmm, "error": None}
+    except Exception as e:
+        return {"success": False, "result": None, "error": str(e)}
+
+@mcp.tool(name="trace_tool", description="Trace a model execution")
+def trace_tool(model: callable, data: dict) -> dict:
+    """
+    Trace the execution of a model with the provided data.
+
+    Parameters:
+    - model: The probabilistic model function.
+    - data: A dictionary containing observed data.
+
+    Returns:
+    - A dictionary with success, result, and error fields.
+    """
+    try:
+        traced = trace(model).get_trace(data)
+        return {"success": True, "result": traced, "error": None}
+    except Exception as e:
+        return {"success": False, "result": None, "error": str(e)}
+
+def create_app() -> FastMCP:
+    """
+    Create and return the FastMCP application instance.
+
+    Returns:
+    - The FastMCP application instance.
+    """
+    return mcp

pyro/mcp_output/requirements.txt

@@ -0,0 +1,9 @@
+fastmcp
+fastapi
+uvicorn[standard]
+pydantic>=2.0.0
+numpy>=1.7
+opt_einsum>=2.3.2
+pyro-api>=0.1.1
+torch>=2.0
+tqdm>=4.36

pyro/mcp_output/start_mcp.py

@@ -0,0 +1,30 @@
+
+"""
+MCP Service Startup Entry
+"""
+import sys
+import os
+
+project_root = os.path.dirname(os.path.abspath(__file__))
+mcp_plugin_dir = os.path.join(project_root, "mcp_plugin")
+if mcp_plugin_dir not in sys.path:
+    sys.path.insert(0, mcp_plugin_dir)
+
+from mcp_service import create_app
+
+def main():
+    """Start FastMCP service"""
+    app = create_app()
+    # Use environment variable to configure port, default 8000
+    port = int(os.environ.get("MCP_PORT", "8000"))
+    
+    # Choose transport mode based on environment variable
+    transport = os.environ.get("MCP_TRANSPORT", "stdio")
+    if transport == "http":
+        app.run(transport="http", host="0.0.0.0", port=port)
+    else:
+        # Default to STDIO mode
+        app.run()
+
+if __name__ == "__main__":
+    main()

pyro/mcp_output/workflow_summary.json

@@ -0,0 +1,219 @@
+{
+  "repository": {
+    "name": "pyro",
+    "url": "https://github.com/pyro-ppl/pyro",
+    "local_path": "/export/zxcpu1/shiweijie/code/ghh/Code2MCP/workspace/pyro",
+    "description": "Python library",
+    "features": "Basic functionality",
+    "tech_stack": "Python",
+    "stars": 0,
+    "forks": 0,
+    "language": "Python",
+    "last_updated": "",
+    "complexity": "medium",
+    "intrusiveness_risk": "low"
+  },
+  "execution": {
+    "start_time": 1770447584.6202667,
+    "end_time": 1770447708.0682755,
+    "duration": 123.44800877571106,
+    "status": "success",
+    "workflow_status": "success",
+    "nodes_executed": [
+      "download",
+      "analysis",
+      "env",
+      "generate",
+      "run",
+      "review",
+      "finalize"
+    ],
+    "total_files_processed": 26,
+    "environment_type": "unknown",
+    "llm_calls": 0,
+    "deepwiki_calls": 0
+  },
+  "tests": {
+    "original_project": {
+      "passed": false,
+      "details": {},
+      "test_coverage": "100%",
+      "execution_time": 0,
+      "test_files": []
+    },
+    "mcp_plugin": {
+      "passed": true,
+      "details": {},
+      "service_health": "healthy",
+      "startup_time": 0,
+      "transport_mode": "stdio",
+      "fastmcp_version": "unknown",
+      "mcp_version": "unknown"
+    }
+  },
+  "analysis": {
+    "structure": {
+      "packages": [
+        "source.examples",
+        "source.examples.contrib",
+        "source.examples.cvae",
+        "source.examples.mixed_hmm",
+        "source.examples.scanvi",
+        "source.profiler",
+        "source.pyro",
+        "source.pyro.contrib",
+        "source.pyro.distributions",
+        "source.pyro.infer",
+        "source.pyro.nn",
+        "source.pyro.ops",
+        "source.pyro.optim",
+        "source.pyro.params",
+        "source.pyro.poutine",
+        "source.tests",
+        "source.tests.contrib",
+        "source.tests.distributions",
+        "source.tests.infer",
+        "source.tests.integration_tests",
+        "source.tests.nn",
+        "source.tests.ops",
+        "source.tests.optim",
+        "source.tests.params",
+        "source.tests.perf",
+        "source.tests.poutine"
+      ]
+    },
+    "dependencies": {
+      "has_environment_yml": false,
+      "has_requirements_txt": false,
+      "pyproject": true,
+      "setup_cfg": true,
+      "setup_py": true
+    },
+    "entry_points": {
+      "imports": [],
+      "cli": [],
+      "modules": []
+    },
+    "risk_assessment": {
+      "import_feasibility": 0.9,
+      "intrusiveness_risk": "low",
+      "complexity": "medium"
+    },
+    "deepwiki_analysis": {
+      "repo_url": "https://github.com/pyro-ppl/pyro",
+      "repo_name": "pyro",
+      "content": "pyro-ppl/pyro\nCore Primitives and Effect Handlers\nEffect Handlers\nTrace System\nDistributions\nDistribution Classes\nTransforms and Constraints\nSpecial Distributions\nDistribution Utilities\nInference Algorithms\nStochastic Variational Inference\nELBO Variants\nMCMC Methods\nImportance Sampling and Predictive\nModels and Guides\nThe Model/Guide Pattern\nParameter Management\nParameter Store\nExamples and Applications\nTutorial System\nBayesian Statistical Models\nDeep Generative Models\nSpecialized Applications\nDevelopment Infrastructure\nSetup and Installation\nTesting Framework\nSpecial Operations and Utilities\ndocs/requirements.txt\ndocs/source/conf.py\npyproject.toml\npyro/__init__.py\npyro/distributions/constraints.py\npyro/poutine/__init__.py\npyro/util.py\nThis page provides a high-level introduction to Pyro's architecture, design philosophy, and core components. It serves as an entry point for understanding how the system is structured and how its parts interact.\nFor detailed information about specific subsystems, see:\nCore primitives and effect handlers:Section 2\nDistribution system:Section 3\nInference algorithms:Section 4\nModel and guide patterns:Section 5\nParameter management:Section 6\nWhat is Pyro?\nPyro is a universal probabilistic programming language (PPL) built on PyTorch. It enables users to write generative models as Python programs and perform inference over them using a variety of algorithms including variational inference (SVI) and Markov chain Monte Carlo (MCMC).\nPackage Identity:\nName:pyro-pplsetup.py86\nDescription:\"A Python library for probabilistic modeling and inference\"setup.py88\nVersion:1.9.1+pyro/__init__.py31\nPython Requirements:>=3.8setup.py147\nCore Dependencies:\nSources:setup.py102-111pyro/__init__.py1-67\nDesign Philosophy\nPyro is built on four core principles that guide its architecture:\nUniversalAny computable distributionScalableMinimal overhead vs hand-written codeMinimalSmall core, powerful abstractionsFlexibleAutomation + manual controlPyro Core System\nUniversalAny computable distribution\nScalableMinimal overhead vs hand-written code\nMinimalSmall core, powerful abstractions\nFlexibleAutomation + manual control\nPyro Core System\nPyro can represent any computable probability distribution through its flexible programming model. Models are ordinary Python functions that can include loops, recursion, and arbitrary control flow.\nPyro is designed to scale to large datasets with minimal overhead. It builds directly on PyTorch's tensor operations and automatic differentiation, avoiding unnecessary abstractions.\nThe core system is implemented with a small set of composable abstractions. The effect handler system (poutine) provides the foundation for model manipulation without requiring extensive framework code.\nPyro provides high-level automation (e.g., automatic variational guides) while allowing experts to customize inference at any level through effect handlers and custom implementations.\nSources:README.md24-29\nSystem Architecture Overview\nPyro's architecture consists of several layered subsystems:\nParameter LayerDistribution LayerInference LayerEffect Handler Layer (Poutine)Primitive LayerUser LayerUser Model CodePython functions withpyro.sample, pyro.paramPyroModuleNeural network integrationExamples & TutorialsVAE, GP, HMM, etc.Primitive Operationspyro.samplepyro.parampyro.platepyro.factorMessage Runtime_PYRO_STACKapply_stack()Effect Handlerstrace, replaycondition, blockscale, enumSVIStochastic Variational InferenceMCMCHMC, NUTSAutoGuideAutomatic guide generationELBO VariantsTrace_ELBOTraceEnum_ELBOPyTorch Distributionstorch.distributions.*Pyro DistributionsDelta, HMM, etc.Constraints & TransformsParamStoreDictGlobal parameter storagePyroOptimWrapped PyTorch optimizers\nParameter Layer\nDistribution Layer\nInference Layer\nEffect Handler Layer (Poutine)\nPrimitive Layer\nUser Model CodePython functions withpyro.sample, pyro.param\nPyroModuleNeural network integration\nExamples & TutorialsVAE, GP, HMM, etc.\nPrimitive Operationspyro.samplepyro.parampyro.platepyro.factor\nMessage Runtime_PYRO_STACKapply_stack()\nEffect Handlerstrace, replaycondition, blockscale, enum\nSVIStochastic Variational Inference\nMCMCHMC, NUTS\nAutoGuideAutomatic guide generation\nELBO VariantsTrace_ELBOTraceEnum_ELBO\nPyTorch Distributionstorch.distributions.*\nPyro DistributionsDelta, HMM, etc.\nConstraints & Transforms\nParamStoreDictGlobal parameter storage\nPyroOptimWrapped PyTorch optimizers\nSources:pyro/__init__.py1-67pyro/poutine/__init__.py1-59\nCore Components and Code Entities\nPrimitives Module\nThe top-level module exports the fundamental primitives that users interact with:\npyro.sample\npyro.primitives\npyro.primitives\npyro.primitives\npyro.factor\npyro.primitives\npyro.deterministic\npyro.primitives\nAdditional utilities:\npyro.enable_validation()- Enable shape and distribution validationpyro/__init__.py12\npyro.enable_validation()\npyro.clear_param_store()- Reset global parameter storagepyro/__init__.py10\npyro.clear_param_store()\npyro.get_param_store()- Access the global parameter storepyro/__init__.py14\npyro.get_param_store()\npyro.set_rng_seed()- Set random number generator seedpyro/util.py37-45\npyro.set_rng_seed()\nSources:pyro/__init__.py8-26pyro/__init__.py39-66\nPoutine (Effect Handler) System\nPoutine is the effect handler system that intercepts and transforms probabilistic primitives:\nTrace Data StructureCore RuntimeHandler ExportstraceRecords executionreplayReplays recorded samplesconditionFixes observationsblockHides sites from handlersscaleScales log probabilitiesenumEnumerates discrete variablesreparamApplies reparameterization_PYRO_STACKGlobal message stackapply_stack()Message dispatcherMessengerBase handler classTraceExecution recordingMessageSite information dict\nTrace Data Structure\nCore Runtime\nHandler Exports\ntraceRecords execution\nreplayReplays recorded samples\nconditionFixes observations\nblockHides sites from handlers\nscaleScales log probabilities\nenumEnumerates discrete variables\nreparamApplies reparameterization\n_PYRO_STACKGlobal message stack\napply_stack()Message dispatcher\nMessengerBase handler class\nTraceExecution recording\nMessageSite information dict\nKey handler exports frompyro.poutine:\npyro.poutine\ntrace- Records model execution into a Trace objectpyro/poutine/__init__.py23\nreplay- Re-executes model with fixed sample valuespyro/poutine/__init__.py19\ncondition- Conditions model on observed datapyro/poutine/__init__.py8\nblock- Blocks primitives from outer handlerspyro/poutine/__init__.py5\nscale- Scales log probabilities (for subsampling)pyro/poutine/__init__.py20\nenum- Enumerates discrete latent variablespyro/poutine/__init__.py10\nSources:pyro/poutine/__init__.py1-59\nProgramming Model: From Code to Execution\nThis diagram shows how user code flows through Pyro's execution model:\n\"ParamStoreDict\"\"Distribution\"\"Effect Handlers\"\"_PYRO_STACK\"\"pyro.sample()\"\"User Code\"\"ParamStoreDict\"\"Distribution\"\"Effect Handlers\"\"_PYRO_STACK\"\"pyro.sample()\"\"User Code\"Message contains:name, fn, value, args, kwargsEach handler transforms messagePost-process (top-to-bottom)alt[If pyro.param() called]pyro.sample(\"x\", dist)Create messageProcess message (bottom-to-top)_process_message()Sample from distributionSample value_postprocess_message()Return processed messagevalueReturn samplepyro.param(\"theta\")Get/create parameterParameter tensorReturn parameter\nSources:pyro/__init__.py8-26pyro/poutine/__init__.py27-28\nParameter Management Architecture\nPyro maintains a global parameter store that synchronizes with PyTorch module parameters:\nOptimizationGlobal StorageUser Interfacepyro.param('theta', init)Get/create parameterPyroModule.x = PyroParam(...)Module-level parameterParamStoreDict_params: Dict[str, Tensor]_constraints: Dict[str, Constraint]pyro.get_param_store()Access singletonpyro.clear_param_store()Reset all parametersPyroOptimWraps torch.optimSVI.step()Optimization step\nOptimization\nGlobal Storage\nUser Interface\npyro.param('theta', init)Get/create parameter\nPyroModule.x = PyroParam(...)Module-level parameter\nParamStoreDict_params: Dict[str, Tensor]_constraints: Dict[str, Constraint]\npyro.get_param_store()Access singleton\npyro.clear_param_store()Reset all parameters\nPyroOptimWraps torch.optim\nSVI.step()Optimization step\nKey operations:\npyro.param(name, init_value, constraint=constraints.real)- Declare/retrieve parameter\npyro.param(name, init_value, constraint=constraints.real)\npyro.get_param_store()- Get the global ParamStoreDict instancepyro/__init__.py14\npyro.get_param_store()\npyro.clear_param_store()- Clear all parameterspyro/__init__.py10\npyro.clear_param_store()\nSources:pyro/__init__.py10-18pyro/__init__.py54\nDistribution System\nPyro extends PyTorch's distribution library with additional distributions and functionality:\nPyro-SpecificStandard DistributionsPyro ExtensionsPyTorch Basetorch.distributions.Distributiontorch.distributions.constraintsTorchDistributionMixinPyro compatibility layerpyro.distributions.constraintsAdditional constraintsNormal, Beta, Gamma, etc.Wrapped PyTorch distributionsDeltaDegenerate distributionHiddenMarkovModelDiscreteHMM, GaussianHMMEmpiricalSample-based distribution\nPyro-Specific\nStandard Distributions\nPyro Extensions\nPyTorch Base\ntorch.distributions.Distribution\ntorch.distributions.constraints\nTorchDistributionMixinPyro compatibility layer\npyro.distributions.constraintsAdditional constraints\nNormal, Beta, Gamma, etc.Wrapped PyTorch distributions\nDeltaDegenerate distribution\nHiddenMarkovModelDiscreteHMM, GaussianHMM\nEmpiricalSample-based distribution\nConstraint extensionsinpyro.distributions.constraints:\npyro.distributions.constraints\ninteger- Integer valuespyro/distributions/constraints.py152\nsphere- Points on Euclidean spherepyro/distributions/constraints.py155\ncorr_matrix- Correlation matricespyro/distributions/constraints.py151\ncorr_matrix\nordered_vector- Monotonically increasing vectorspyro/distributions/constraints.py153\nordered_vector\nAll PyTorch constraints viafrom torch.distributions.constraints import *pyro/distributions/constraints.py5\nfrom torch.distributions.constraints import *\nSources:pyro/distributions/constraints.py1-235\nInference Ecosystem\nPyro provides multiple inference algorithms that operate on the same model-guide abstraction:\npyro.infer.SVI\npyro.infer.MCMC\npyro.infer.MCMC\npyro.infer.Importance\nELBO variants for SVI:\nTrace_ELBO- Basic Monte Carlo ELBO estimator\nTraceEnum_ELBO- Enumerates discrete latent variables\nTraceEnum_ELBO\nTraceGraph_ELBO- Variance reduction via Rao-Blackwellization\nTraceGraph_ELBO\nRenyiELBO- Rényi divergence objective (α-divergence)\nAutomatic guides:",
+      "model": "gpt-4o-2024-08-06",
+      "source": "selenium",
+      "success": true
+    },
+    "code_complexity": {
+      "cyclomatic_complexity": "medium",
+      "cognitive_complexity": "medium",
+      "maintainability_index": 75
+    },
+    "security_analysis": {
+      "vulnerabilities_found": 0,
+      "security_score": 85,
+      "recommendations": []
+    }
+  },
+  "plugin_generation": {
+    "files_created": [
+      "mcp_output/start_mcp.py",
+      "mcp_output/mcp_plugin/__init__.py",
+      "mcp_output/mcp_plugin/mcp_service.py",
+      "mcp_output/mcp_plugin/adapter.py",
+      "mcp_output/mcp_plugin/main.py",
+      "mcp_output/requirements.txt",
+      "mcp_output/README_MCP.md"
+    ],
+    "main_entry": "start_mcp.py",
+    "requirements": [
+      "fastmcp>=0.1.0",
+      "pydantic>=2.0.0"
+    ],
+    "readme_path": "/export/zxcpu1/shiweijie/code/ghh/Code2MCP/workspace/pyro/mcp_output/README_MCP.md",
+    "adapter_mode": "import",
+    "total_lines_of_code": 0,
+    "generated_files_size": 0,
+    "tool_endpoints": 0,
+    "supported_features": [
+      "Basic functionality"
+    ],
+    "generated_tools": [
+      "Basic tools",
+      "Health check tools",
+      "Version info tools"
+    ]
+  },
+  "code_review": {},
+  "errors": [],
+  "warnings": [],
+  "recommendations": [
+    "Improve test coverage by adding more unit tests for critical modules",
+    "streamline the import strategy to reduce complexity",
+    "enhance documentation for better understanding of core components",
+    "optimize the size of large files for better performance",
+    "ensure all dependencies are clearly defined and up-to-date",
+    "implement continuous integration to automate testing and deployment",
+    "refactor code to improve readability and maintainability",
+    "consider adding more examples and tutorials to demonstrate usage",
+    "review and update the README file to provide a comprehensive overview of the project",
+    "conduct regular code reviews to maintain code quality."
+  ],
+  "performance_metrics": {
+    "memory_usage_mb": 0,
+    "cpu_usage_percent": 0,
+    "response_time_ms": 0,
+    "throughput_requests_per_second": 0
+  },
+  "deployment_info": {
+    "supported_platforms": [
+      "Linux",
+      "Windows",
+      "macOS"
+    ],
+    "python_versions": [
+      "3.8",
+      "3.9",
+      "3.10",
+      "3.11",
+      "3.12"
+    ],
+    "deployment_methods": [
+      "Docker",
+      "pip",
+      "conda"
+    ],
+    "monitoring_support": true,
+    "logging_configuration": "structured"
+  },
+  "execution_analysis": {
+    "success_factors": [
+      "Comprehensive node execution covering download, analysis, environment setup, generation, and review",
+      "Successful MCP plugin generation with healthy service status"
+    ],
+    "failure_reasons": [],
+    "overall_assessment": "excellent",
+    "node_performance": {
+      "download_time": "Efficient download process completed successfully",
+      "analysis_time": "Thorough analysis completed within expected timeframe",
+      "generation_time": "Code generation executed smoothly without delays",
+      "test_time": "Testing phase completed quickly, but original project tests did not pass"
+    },
+    "resource_usage": {
+      "memory_efficiency": "Memory usage data not available",
+      "cpu_efficiency": "CPU usage data not available",
+      "disk_usage": "Disk usage data not available"
+    }
+  },
+  "technical_quality": {
+    "code_quality_score": 85,
+    "architecture_score": 90,
+    "performance_score": 80,
+    "maintainability_score": 75,
+    "security_score": 85,
+    "scalability_score": 80
+  }
+}

pyro/source/.codecov.yml

@@ -0,0 +1,14 @@
+# Copyright Contributors to the Pyro project.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+ignore:
+  - "pyro/docutil.py"
+  - "pyro/logger.py"
+
+coverage:
+  range: 60..95
+  round: nearest
+  precision: 2
+
+comment: false

pyro/source/.coveragerc

@@ -0,0 +1,21 @@
+# Copyright Contributors to the Pyro project.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+[report]
+omit =
+    pyro/docutil.py
+    pyro/logger.py
+exclude_lines =
+    pragma: no cover
+    def backward
+    raise AssertionError
+    raise NotImplementedError
+    raise ValueError
+    except NotImplementedError
+    except ImportError
+    except KeyError
+    except TypeError
+    warnings\.warn.*
+    warn_if.*
+    if __name__ == .__main__.:

pyro/source/.readthedocs.yml

@@ -0,0 +1,23 @@
+# Copyright Contributors to the Pyro project.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+# Required
+version: 2
+
+build:
+  os: ubuntu-20.04
+  tools:
+    python: "3.8"
+
+# Build documentation in the docs/ directory with Sphinx
+sphinx:
+  configuration: docs/source/conf.py
+
+# Optionally build your docs in additional formats such as PDF and ePub
+# formats: all
+
+# Optionally set the version of Python and requirements required to build your docs
+python:
+  install:
+    - requirements: docs/requirements.txt

pyro/source/CODE_OF_CONDUCT.md

@@ -0,0 +1,53 @@
+
+<!--
+Copyright Contributors to the Pyro project.
+
+SPDX-License-Identifier: Apache-2.0
+-->
+
+# Contributor Covenant Code of Conduct
+
+## Our Pledge
+
+In the interest of fostering an open and welcoming environment, we as contributors and maintainers pledge to making participation in our project and our community a harassment-free experience for everyone, regardless of age, body size, disability, ethnicity, gender identity and expression, level of experience, nationality, personal appearance, race, religion, or sexual identity and orientation.
+
+## Our Standards
+
+Examples of behavior that contributes to creating a positive environment include:
+
+* Using welcoming and inclusive language
+* Being respectful of differing viewpoints and experiences
+* Gracefully accepting constructive criticism
+* Focusing on what is best for the community
+* Showing empathy towards other community members
+
+Examples of unacceptable behavior by participants include:
+
+* The use of sexualized language or imagery and unwelcome sexual attention or advances
+* Trolling, insulting/derogatory comments, and personal or political attacks
+* Public or private harassment
+* Publishing others' private information, such as a physical or electronic address, without explicit permission
+* Other conduct which could reasonably be considered inappropriate in a professional setting
+
+## Our Responsibilities
+
+Project maintainers are responsible for clarifying the standards of acceptable behavior and are expected to take appropriate and fair corrective action in response to any instances of unacceptable behavior.
+
+Project maintainers have the right and responsibility to remove, edit, or reject comments, commits, code, wiki edits, issues, and other contributions that are not aligned to this Code of Conduct, or to ban temporarily or permanently any contributor for other behaviors that they deem inappropriate, threatening, offensive, or harmful.
+
+## Scope
+
+This Code of Conduct applies both within project spaces and in public spaces when an individual is representing the project or its community. Examples of representing a project or community include using an official project e-mail address, posting via an official social media account, or acting as an appointed representative at an online or offline event. Representation of a project may be further defined and clarified by project maintainers.
+
+## Enforcement
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be reported by contacting the project team at fritz.obermeyer@gmail.com or fehiepsi@gmail.com. The project team will review and investigate all complaints, and will respond in a way that it deems appropriate to the circumstances. The project team is obligated to maintain confidentiality with regard to the reporter of an incident. Further details of specific enforcement policies may be posted separately.
+
+Project maintainers who do not follow or enforce the Code of Conduct in good faith may face temporary or permanent repercussions as determined by other members of the project's leadership.
+
+## Attribution
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage], version 1.4, available at [http://contributor-covenant.org/version/1/4][version]
+
+[homepage]: http://contributor-covenant.org
+[version]: http://contributor-covenant.org/version/1/4/

pyro/source/CONTRIBUTING.md

@@ -0,0 +1,133 @@
+<!--
+Copyright Contributors to the Pyro project.
+
+SPDX-License-Identifier: Apache-2.0
+-->
+
+# Development
+
+Please follow our established coding style including variable names, module imports, and function definitions.
+The Pyro codebase follows the [PEP8 style guide](https://www.python.org/dev/peps/pep-0008/)
+(which you can check with `make lint`) and follows
+[`isort`](https://github.com/timothycrosley/isort) import order (which you can enforce with `make format`).
+When creating new files please add a license header; this can be done automatically via `make license` or simply `make format`.
+
+# Setup
+
+First install [PyTorch](http://pytorch.org/).
+
+Then, install all the dev dependencies for Pyro.
+```sh
+make install
+```
+or explicitly
+```sh
+pip install -e .[dev]
+```
+
+# Testing
+
+Before submitting a pull request, please autoformat code and ensure that unit tests pass locally
+```sh
+make format            # runs isort
+make test              # linting and unit tests
+```
+
+If you've modified core pyro code, examples, or tutorials, you can run more comprehensive tests locally (after first adding any new files to the appropriate `tests/` script)
+```sh
+make test-examples     # test examples/
+make integration-test  # longer-running tests (may take hours)
+make test-cuda         # runs unit tests in cuda mode
+```
+
+To run all tests locally in parallel, use the `pytest-xdist` package
+```sh
+pip install pytest-xdist
+pytest -vs -n auto
+```
+
+To run a single test from the command line
+```sh
+pytest -vs {path_to_test}::{test_name}
+# or in cuda mode
+CUDA_TEST=1 PYRO_DTYPE=float64 PYRO_DEVICE=cuda pytest -vs {path_to_test}::{test_name}
+```
+
+To ensure documentation builds correctly, run
+```sh
+make docs
+```
+
+## Testing Tutorials
+
+We run some tutorials on travis to avoid bit rot.
+Before submitting a new tutorial, please run `make scrub` from 
+the top-level pyro directory in order to scrub the metadata in 
+the notebooks.
+To enable a tutorial for testing
+
+1.  Add a line `smoke_test = ('CI' in os.environ)` to your tutorial. Our test
+    scripts only test tutorials that contain the string `smoke_test`.
+2.  Each time you do something expensive for many iterations, set the number
+    of iterations like this:
+    ```py
+    for epoch in range(200 if not smoke_test else 1):
+        ...
+    ```
+
+You can test locally by running `make test-tutorials`.
+
+# Profiling
+
+The profiler module contains scripts to support profiling different 
+Pyro modules, as well as test for performance regression.
+
+To run the profiling utilities, ensure that all dependencies for profiling are satisfied, 
+by running `make install`, or more specifically, `pip install -e .[profile]`.
+
+There are some generic test cases available in the `profiler` module. Currently, this supports 
+only the `distributions` library, but we will be adding test cases for inference methods
+soon.
+
+#### Some useful invocations
+
+To get help on the parameters that the profiling script takes, run: 
+
+```sh
+python -m profiler.distributions --help
+```
+
+To run the profiler on all the distributions, simply run:
+
+```sh
+python -m profiler.distributions
+```
+
+To run the profiler on a few distributions by varying the batch size, run:
+
+```sh
+python -m profiler.distributions --dist bernoulli normal --batch_sizes 1000 100000 
+```
+
+To get more details on the potential sources of slowdown, use the `cProfile` tool
+ as follows:
+
+```sh
+python -m profiler.distributions --dist bernoulli --tool cprofile
+```
+
+# Submitting
+
+For larger changes, please open an issue for discussion before submitting a pull request.
+For relevant design questions to consider, see past
+[design documents](https://github.com/pyro-ppl/pyro/wiki/Design-Docs).
+
+In your pull request description on github, please note:
+- Proposed changes
+- Links to related issues/PRs
+- New and existing tests
+
+Before submitting, please run `make format`, `make lint`, and running tests as described above.
+
+For speculative changes meant for early-stage review, include `[WIP]` in the PR's title. 
+(One of the maintainers will add the `WIP` tag.)

pyro/source/LICENSE.md

@@ -0,0 +1,202 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

pyro/source/LICENSES/Apache-2.0.txt

@@ -0,0 +1,73 @@
+Apache License
+Version 2.0, January 2004
+http://www.apache.org/licenses/
+
+TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+1. Definitions.
+
+"License" shall mean the terms and conditions for use, reproduction, and distribution as defined by Sections 1 through 9 of this document.
+
+"Licensor" shall mean the copyright owner or entity authorized by the copyright owner that is granting the License.
+
+"Legal Entity" shall mean the union of the acting entity and all other entities that control, are controlled by, or are under common control with that entity. For the purposes of this definition, "control" means (i) the power, direct or indirect, to cause the direction or management of such entity, whether by contract or otherwise, or (ii) ownership of fifty percent (50%) or more of the outstanding shares, or (iii) beneficial ownership of such entity.
+
+"You" (or "Your") shall mean an individual or Legal Entity exercising permissions granted by this License.
+
+"Source" form shall mean the preferred form for making modifications, including but not limited to software source code, documentation source, and configuration files.
+
+"Object" form shall mean any form resulting from mechanical transformation or translation of a Source form, including but not limited to compiled object code, generated documentation, and conversions to other media types.
+
+"Work" shall mean the work of authorship, whether in Source or Object form, made available under the License, as indicated by a copyright notice that is included in or attached to the work (an example is provided in the Appendix below).
+
+"Derivative Works" shall mean any work, whether in Source or Object form, that is based on (or derived from) the Work and for which the editorial revisions, annotations, elaborations, or other modifications represent, as a whole, an original work of authorship. For the purposes of this License, Derivative Works shall not include works that remain separable from, or merely link (or bind by name) to the interfaces of, the Work and Derivative Works thereof.
+
+"Contribution" shall mean any work of authorship, including the original version of the Work and any modifications or additions to that Work or Derivative Works thereof, that is intentionally submitted to Licensor for inclusion in the Work by the copyright owner or by an individual or Legal Entity authorized to submit on behalf of the copyright owner. For the purposes of this definition, "submitted" means any form of electronic, verbal, or written communication sent to the Licensor or its representatives, including but not limited to communication on electronic mailing lists, source code control systems, and issue tracking systems that are managed by, or on behalf of, the Licensor for the purpose of discussing and improving the Work, but excluding communication that is conspicuously marked or otherwise designated in writing by the copyright owner as "Not a Contribution."
+
+"Contributor" shall mean Licensor and any individual or Legal Entity on behalf of whom a Contribution has been received by Licensor and subsequently incorporated within the Work.
+
+2. Grant of Copyright License. Subject to the terms and conditions of this License, each Contributor hereby grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable copyright license to reproduce, prepare Derivative Works of, publicly display, publicly perform, sublicense, and distribute the Work and such Derivative Works in Source or Object form.
+
+3. Grant of Patent License. Subject to the terms and conditions of this License, each Contributor hereby grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable (except as stated in this section) patent license to make, have made, use, offer to sell, sell, import, and otherwise transfer the Work, where such license applies only to those patent claims licensable by such Contributor that are necessarily infringed by their Contribution(s) alone or by combination of their Contribution(s) with the Work to which such Contribution(s) was submitted. If You institute patent litigation against any entity (including a cross-claim or counterclaim in a lawsuit) alleging that the Work or a Contribution incorporated within the Work constitutes direct or contributory patent infringement, then any patent licenses granted to You under this License for that Work shall terminate as of the date such litigation is filed.
+
+4. Redistribution. You may reproduce and distribute copies of the Work or Derivative Works thereof in any medium, with or without modifications, and in Source or Object form, provided that You meet the following conditions:
+
+     (a) You must give any other recipients of the Work or Derivative Works a copy of this License; and
+
+     (b) You must cause any modified files to carry prominent notices stating that You changed the files; and
+
+     (c) You must retain, in the Source form of any Derivative Works that You distribute, all copyright, patent, trademark, and attribution notices from the Source form of the Work, excluding those notices that do not pertain to any part of the Derivative Works; and
+
+     (d) If the Work includes a "NOTICE" text file as part of its distribution, then any Derivative Works that You distribute must include a readable copy of the attribution notices contained within such NOTICE file, excluding those notices that do not pertain to any part of the Derivative Works, in at least one of the following places: within a NOTICE text file distributed as part of the Derivative Works; within the Source form or documentation, if provided along with the Derivative Works; or, within a display generated by the Derivative Works, if and wherever such third-party notices normally appear. The contents of the NOTICE file are for informational purposes only and do not modify the License. You may add Your own attribution notices within Derivative Works that You distribute, alongside or as an addendum to the NOTICE text from the Work, provided that such additional attribution notices cannot be construed as modifying the License.
+
+     You may add Your own copyright statement to Your modifications and may provide additional or different license terms and conditions for use, reproduction, or distribution of Your modifications, or for any such Derivative Works as a whole, provided Your use, reproduction, and distribution of the Work otherwise complies with the conditions stated in this License.
+
+5. Submission of Contributions. Unless You explicitly state otherwise, any Contribution intentionally submitted for inclusion in the Work by You to the Licensor shall be under the terms and conditions of this License, without any additional terms or conditions. Notwithstanding the above, nothing herein shall supersede or modify the terms of any separate license agreement you may have executed with Licensor regarding such Contributions.
+
+6. Trademarks. This License does not grant permission to use the trade names, trademarks, service marks, or product names of the Licensor, except as required for reasonable and customary use in describing the origin of the Work and reproducing the content of the NOTICE file.
+
+7. Disclaimer of Warranty. Unless required by applicable law or agreed to in writing, Licensor provides the Work (and each Contributor provides its Contributions) on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied, including, without limitation, any warranties or conditions of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A PARTICULAR PURPOSE. You are solely responsible for determining the appropriateness of using or redistributing the Work and assume any risks associated with Your exercise of permissions under this License.
+
+8. Limitation of Liability. In no event and under no legal theory, whether in tort (including negligence), contract, or otherwise, unless required by applicable law (such as deliberate and grossly negligent acts) or agreed to in writing, shall any Contributor be liable to You for damages, including any direct, indirect, special, incidental, or consequential damages of any character arising as a result of this License or out of the use or inability to use the Work (including but not limited to damages for loss of goodwill, work stoppage, computer failure or malfunction, or any and all other commercial damages or losses), even if such Contributor has been advised of the possibility of such damages.
+
+9. Accepting Warranty or Additional Liability. While redistributing the Work or Derivative Works thereof, You may choose to offer, and charge a fee for, acceptance of support, warranty, indemnity, or other liability obligations and/or rights consistent with this License. However, in accepting such obligations, You may act only on Your own behalf and on Your sole responsibility, not on behalf of any other Contributor, and only if You agree to indemnify, defend, and hold each Contributor harmless for any liability incurred by, or claims asserted against, such Contributor by reason of your accepting any such warranty or additional liability.
+
+END OF TERMS AND CONDITIONS
+
+APPENDIX: How to apply the Apache License to your work.
+
+To apply the Apache License to your work, attach the following boilerplate notice, with the fields enclosed by brackets "[]" replaced with your own identifying information. (Don't include the brackets!)  The text should be enclosed in the appropriate comment syntax for the file format. We also recommend that a file or class name and description of purpose be included on the same "printed page" as the copyright notice for easier identification within third-party archives.
+
+Copyright [yyyy] [name of copyright owner]
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

pyro/source/LICENSES/BSD-3-Clause.txt

@@ -0,0 +1,11 @@
+Copyright (c) <year> <owner>. 
+
+Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

pyro/source/LICENSES/MIT.txt

@@ -0,0 +1,9 @@
+MIT License
+
+Copyright (c) <year> <copyright holders>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

pyro/source/MANIFEST.in

@@ -0,0 +1,2 @@
+include LICENSE.md MANIFEST.in
+recursive-include pyro *.cpp

pyro/source/Makefile

@@ -0,0 +1,91 @@
+.PHONY: all install docs lint format test integration-test clean FORCE
+
+all: docs test
+
+install: FORCE
+	pip install -e .[dev,profile] --config-settings editable_mode=strict
+
+uninstall: FORCE
+	pip uninstall pyro-ppl
+
+docs: FORCE
+	$(MAKE) -C docs html
+
+apidoc: FORCE
+	$(MAKE) -C docs apidoc
+
+tutorial: FORCE
+	$(MAKE) -C tutorial html
+
+lint: FORCE
+	ruff check .
+	black --check *.py pyro examples tests scripts profiler
+	python scripts/update_headers.py --check
+	mypy --install-types --non-interactive pyro scripts tests
+
+license: FORCE
+	python scripts/update_headers.py
+
+format: license FORCE
+	ruff check --fix .
+	black *.py pyro examples tests scripts profiler
+
+version: FORCE
+	python scripts/update_version.py
+
+scrub: FORCE
+	find tutorial -name "*.ipynb" | xargs python -m nbstripout --keep-output --keep-count
+	find tutorial -name "*.ipynb" | xargs python tutorial/source/cleannb.py
+
+doctest: FORCE
+	# We skip testing pyro.distributions.torch wrapper classes because
+	# they include torch docstrings which are tested upstream.
+	python -m pytest -p tests.doctest_fixtures --doctest-modules -o filterwarnings=ignore pyro --ignore=pyro/distributions/torch.py
+
+perf-test: FORCE
+	bash scripts/perf_test.sh ${ref}
+
+profile: ref=dev
+
+profile: FORCE
+	bash scripts/profile_model.sh ${ref} ${models}
+
+test: lint docs doctest FORCE
+	pytest -vx -n auto --stage unit
+
+test-examples: lint FORCE
+	pytest -vx --stage test_examples
+
+test-tutorials: lint FORCE
+	grep -l smoke_test tutorial/source/*.ipynb | xargs grep -L 'smoke_test = False' \
+		| CI=1 xargs pytest -vx --nbval-lax --current-env
+
+integration-test: lint FORCE
+	pytest -vx -n auto --stage integration
+
+test-all: lint FORCE
+	pytest -vx -n auto
+	CI=1 grep -l smoke_test tutorial/source/*.ipynb \
+	  | xargs pytest -vx --nbval-lax
+
+test-cuda: lint FORCE
+	CUDA_TEST=1 PYRO_DTYPE=float64 PYRO_DEVICE=cuda pytest -vx --stage unit
+	CUDA_TEST=1 pytest -vx tests/test_examples.py::test_cuda
+
+test-cuda-lax: lint FORCE
+	CUDA_TEST=1 PYRO_DTYPE=float64 PYRO_DEVICE=cuda pytest -vx --stage unit --lax
+	CUDA_TEST=1 pytest -vx tests/test_examples.py::test_cuda
+
+test-jit: FORCE
+	@echo See jit.log
+	pytest -v -n auto --tb=short --runxfail tests/infer/test_jit.py tests/test_examples.py::test_jit | tee jit.log
+	pytest -v -n auto --tb=short --runxfail tests/infer/mcmc/test_hmc.py tests/infer/mcmc/test_nuts.py \
+		-k JIT=True | tee -a jit.log
+
+test-funsor: lint FORCE
+	pytest -vx -n auto --stage funsor
+
+clean: FORCE
+	git clean -dfx -e pyro_ppl.egg-info
+
+FORCE:

pyro/source/README.md

@@ -0,0 +1,110 @@
+<!--
+Copyright Contributors to the Pyro project.
+
+SPDX-License-Identifier: Apache-2.0
+-->
+
+<div align="center">
+  <a href="http://pyro.ai"> <img width="220px" height="220px" src="docs/source/_static/img/pyro_logo_with_text.png"></a>
+</div>
+
+-----------------------------------------
+
+[![Build Status](https://github.com/pyro-ppl/pyro/workflows/CI/badge.svg)](https://github.com/pyro-ppl/pyro/actions)
+[![Coverage Status](https://coveralls.io/repos/github/pyro-ppl/pyro/badge.svg?branch=dev)](https://coveralls.io/github/pyro-ppl/pyro?branch=dev)
+[![Latest Version](https://badge.fury.io/py/pyro-ppl.svg)](https://pypi.python.org/pypi/pyro-ppl)
+[![Documentation Status](https://readthedocs.org/projects/pyro-ppl/badge/?version=dev)](http://pyro-ppl.readthedocs.io/en/stable/?badge=dev)
+[![CII Best Practices](https://bestpractices.coreinfrastructure.org/projects/3056/badge)](https://bestpractices.coreinfrastructure.org/projects/3056)
+
+[Getting Started](http://pyro.ai/examples) |
+[Documentation](http://docs.pyro.ai/) |
+[Community](http://forum.pyro.ai/) |
+[Contributing](https://github.com/pyro-ppl/pyro/blob/master/CONTRIBUTING.md)
+
+Pyro is a flexible, scalable deep probabilistic programming library built on PyTorch.  Notably, it was designed with these principles in mind:
+
+- **Universal**: Pyro is a universal PPL - it can represent any computable probability distribution.
+- **Scalable**: Pyro scales to large data sets with little overhead compared to hand-written code.
+- **Minimal**: Pyro is agile and maintainable. It is implemented with a small core of powerful, composable abstractions.
+- **Flexible**: Pyro aims for automation when you want it, control when you need it. This is accomplished through high-level abstractions to express generative and inference models, while allowing experts easy-access to customize inference.
+
+Pyro was originally developed at Uber AI and is now actively maintained by community contributors, including a dedicated team at the [Broad Institute](https://www.broadinstitute.org/).
+In 2019, Pyro [became](https://www.linuxfoundation.org/press-release/2019/02/pyro-probabilistic-programming-language-becomes-newest-lf-deep-learning-project/) a project of the Linux Foundation, a neutral space for collaboration on open source software, open standards, open data, and open hardware.
+
+For more information about the high level motivation for Pyro, check out our [launch blog post](http://eng.uber.com/pyro).
+For additional blog posts, check out work on [experimental design](https://eng.uber.com/oed-pyro-release/) and
+[time-to-event modeling](https://eng.uber.com/modeling-censored-time-to-event-data-using-pyro/) in Pyro.
+
+## Installing
+
+### Installing a stable Pyro release
+
+**Install using pip:**
+```sh
+pip install pyro-ppl
+```
+
+**Install from source:**
+```sh
+git clone git@github.com:pyro-ppl/pyro.git
+cd pyro
+git checkout master  # master is pinned to the latest release
+pip install .
+```
+
+**Install with extra packages:**
+
+To install the dependencies required to run the probabilistic models included in the `examples`/`tutorials` directories, please use the following command:
+```sh
+pip install pyro-ppl[extras] 
+```
+Make sure that the models come from the same release version of the [Pyro source code](https://github.com/pyro-ppl/pyro/releases) as you have installed.
+
+### Installing Pyro dev branch
+
+For recent features you can install Pyro from source.
+
+**Install Pyro using pip:**
+
+```sh
+pip install git+https://github.com/pyro-ppl/pyro.git
+```
+or, with the `extras` dependency to run the probabilistic models included in the `examples`/`tutorials` directories:
+```sh
+pip install git+https://github.com/pyro-ppl/pyro.git#egg=project[extras]
+```
+
+**Install Pyro from source:**
+
+```sh
+git clone https://github.com/pyro-ppl/pyro
+cd pyro
+pip install .  # pip install .[extras] for running models in examples/tutorials
+```
+
+## Running Pyro from a Docker Container
+
+Refer to the instructions [here](docker/README.md).
+
+## Citation
+If you use Pyro, please consider citing:
+```
+@article{bingham2019pyro,
+  author    = {Eli Bingham and
+               Jonathan P. Chen and
+               Martin Jankowiak and
+               Fritz Obermeyer and
+               Neeraj Pradhan and
+               Theofanis Karaletsos and
+               Rohit Singh and
+               Paul A. Szerlip and
+               Paul Horsfall and
+               Noah D. Goodman},
+  title     = {Pyro: Deep Universal Probabilistic Programming},
+  journal   = {J. Mach. Learn. Res.},
+  volume    = {20},
+  pages     = {28:1--28:6},
+  year      = {2019},
+  url       = {http://jmlr.org/papers/v20/18-403.html}
+}
+```

pyro/source/RELEASE-MANAGEMENT.md

@@ -0,0 +1,33 @@
+<!--
+Copyright Contributors to the Pyro project.
+
+SPDX-License-Identifier: Apache-2.0
+-->
+
+# Pyro release management
+
+This describes the process by which versions of Pyro are officially released to the public.
+
+## Versioning
+
+Releases are versioned according to the `version_prefix` constant in [pyro/__init__.py](pyro/__init__.py).
+Pyro releases follow semantic versioning with the following caveats:
+
+- Behavior of documented APIs will remain stable across minor releases, except for bug fixes and features marked EXPERIMENTAL or DEPRECATED.
+- Serialization formats will remain stable across patch releases, but may change across minor releases (e.g. if you save a model in 1.0.0, it will be safe to load it in 1.0.1, but not in 1.1.0).
+- Undocumented APIs, features marked EXPERIMENTAL or DEPRECATED, and anything in `pyro.contrib` may change at any time (though we aim for stability).
+- All deprecated features throw a `FutureWarning` and specify possible work-arounds. Features marked as deprecated will not be maintained, and are likely to be removed in a future release.
+- If you want more stability for a particular feature, [contribute](https://github.com/pyro-ppl/pyro/blob/dev/CONTRIBUTING.md) a unit test.
+
+## Release process
+
+Pyro is released at irregular cadence, typically about 4 times per year.
+
+Releases are managed by:
+- [Fritz Obermeyer](https://github.com/fritzo)
+- [Neeraj Pradhan](https://github.com/neerajprad)
+- [JP Chen](https://github.com/jpchen)
+
+Releases and release notes are published to [github](https://github.com/pyro-ppl/pyro/releases).
+Documentation for is published to [readthedocs](https://docs.pyro.ai).
+Release builds are published to [pypi](https://pypi.org/project/pyro-ppl/).

pyro/source/__init__.py

@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+"""
+pyro Project Package Initialization File
+"""

pyro/source/docker/Dockerfile

@@ -0,0 +1,67 @@
+# Copyright Contributors to the Pyro project.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+ARG base_img=ubuntu:24.04
+FROM ${base_img}
+
+# Optional args
+ARG python_version=3
+ARG pyro_branch=release
+ARG pytorch_whl=cpu
+ARG pytorch_branch=release
+ARG uid=1000
+ARG gid=1000
+ARG ostype=Linux
+ARG pyro_git_url=https://github.com/pyro-ppl/pyro.git
+ARG trust_hosts=no
+
+# Configurable settings
+ENV USER_NAME pyromancer
+ENV CONDA_DIR /opt/conda
+ENV WORK_DIR /home/${USER_NAME}/workspace
+ENV PATH ${CONDA_DIR}/bin:${PATH}
+
+# Install linux utils
+RUN apt-get update && apt-get install -y --no-install-recommends \
+         build-essential \
+         cmake \
+         git \
+         wget \
+         ca-certificates && \
+    rm -rf /var/lib/apt/lists/*
+
+# Change to default user
+RUN bash -c 'if [ ${ostype} == Linux ]; then groupadd -r --gid ${gid} ${USER_NAME}; fi && \
+    useradd -r --create-home --shell /bin/bash --uid ${uid} --gid ${gid} ${USER_NAME}' && \
+    mkdir -p ${CONDA_DIR} ${WORK_DIR} && chown ${USER_NAME} ${CONDA_DIR} ${WORK_DIR}
+USER ${USER_NAME}
+
+# Install conda
+RUN if [ ${trust_hosts} = yes ] ; then WGET_ARGS="--no-check-certificate" ; fi && \
+    wget ${WGET_ARGS} -O ~/miniconda.sh \
+    https://repo.anaconda.com/miniconda/Miniconda${python_version%%.*}-latest-Linux-x86_64.sh && \
+    bash ~/miniconda.sh -f -b -p ${CONDA_DIR} && \
+    rm ~/miniconda.sh
+
+# Trust conda and pip hosts if needed
+RUN if [ ${trust_hosts} = yes ] ; \
+    then \
+        pip config set global.trusted-host "pypi.org files.pythonhosted.org download.pytorch.org" && \
+        conda config --set ssl_verify False ; \
+    fi
+
+# Update python version
+RUN conda install python=${python_version}
+
+# Move to home directory; and copy the install script
+WORKDIR ${WORK_DIR}
+COPY install.sh ${WORK_DIR}/install.sh
+
+# Install python 2/3, PyTorch and Pyro
+RUN cd ${WORK_DIR} && conda update -n base conda -c defaults && bash install.sh
+
+# Run Jupyter notebook
+# (Ref: http://jupyter-notebook.readthedocs.io/en/latest/public_server.html#docker-cmd)
+EXPOSE 8888
+CMD ["jupyter", "notebook", "--port=8888", "--no-browser", "--ip=0.0.0.0"]

pyro/source/docker/Makefile

@@ -0,0 +1,163 @@
+# Copyright Contributors to the Pyro project.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+.PHONY: help create-host-workspace build build-gpu run run-gpu notebook notebook-gpu
+
+DOCKER_FILE=Dockerfile
+BASE_IMG=ubuntu:24.04
+BASE_CUDA_IMG=nvidia/cuda:12.6.3-cudnn-runtime-ubuntu24.04
+DOCKER_CMD=docker
+HOST_WORK_DIR=${HOME}/pyro_docker
+UID=999
+GID=999
+OSTYPE=$(shell uname)
+USER=pyromancer
+DOCKER_WORK_DIR=/home/${USER}/workspace/shared
+pyro_git_url=https://github.com/pyro-ppl/pyro.git
+
+# Optional args
+python_version?=3.12
+pytorch_branch?=release
+pyro_branch?=release
+cmd?=bash
+trust_hosts?="no"
+
+# Determine name of docker image
+build run notebook: img_prefix=pyro-cpu
+build-gpu run-gpu notebook-gpu: img_prefix=pyro-gpu
+build run lab: img_prefix=pyro-cpu
+build-gpu run-gpu lab-gpu: img_prefix=pyro-gpu
+
+ifeq ($(img), )
+	IMG_NAME=${img_prefix}-${pyro_branch}-${pytorch_branch}-${python_version}
+else
+	IMG_NAME=${img}
+endif
+
+help:
+	@fgrep -h "##" ${MAKEFILE_LIST} | fgrep -v fgrep | sed -e 's/##//'
+
+##
+##Available targets:
+##
+
+build: ##
+	## Build a docker image for running Pyro on a CPU.
+	## Args:
+	##   python_version: version of python to use. default - python 3.12
+	##   pytorch_branch: whether to build PyTorch from conda or from source
+	##   (git branch specified by pytorch_branch)
+	##     default - latest pytorch version on the torch python package index
+	##   pyro_branch: whether to use the released Pyro wheel or a git branch.
+	##     default - latest pyro-ppl wheel on pypi
+	##   trust_hosts: If set to yes hosts SSL ceritificates will be trusted
+	##   (might be needed when running begind a firewall)
+	##     default - Verify hosts SSL certificates
+	##
+	${DOCKER_CMD} build -t ${IMG_NAME} \
+	--build-arg base_img=${BASE_IMG} \
+	--build-arg uid=${UID} \
+	--build-arg gid=${GID} \
+	--build-arg ostype=${OSTYPE} \
+	--build-arg python_version=${python_version} \
+	--build-arg pytorch_branch=${pytorch_branch} \
+	--build-arg pyro_git_url=${pyro_git_url} \
+	--build-arg pyro_branch=${pyro_branch} \
+	--build-arg trust_hosts=${trust_hosts} -f ${DOCKER_FILE} .
+
+build-gpu: ##
+	## Build a docker image for running Pyro on a GPU.
+	## Args:
+	##   python_version: version of python to use. default - python 3.12
+	##   pytorch_branch: whether to build PyTorch from conda or from source
+	##   (git branch specified by pytorch_branch)
+	##     default - latest pytorch version on the torch python package index
+	##   pyro_branch: whether to use the released Pyro wheel or a git branch.
+	##     default - latest pyro-ppl wheel on pypi
+	##   trust_hosts: If set to yes hosts SSL ceritificates will be trusted
+	##   (might be needed when running begind a firewall)
+	##     default - Verify hosts SSL certificates
+	##
+	${DOCKER_CMD} build -t ${IMG_NAME} \
+	--build-arg base_img=${BASE_CUDA_IMG} \
+	--build-arg uid=${UID} \
+	--build-arg gid=${GID} \
+	--build-arg ostype=${OSTYPE} \
+	--build-arg pytorch_whl=cu126 \
+	--build-arg python_version=${python_version} \
+	--build-arg pytorch_branch=${pytorch_branch} \
+	--build-arg pyro_git_url=${pyro_git_url} \
+	--build-arg pyro_branch=${pyro_branch} \
+	--build-arg trust_hosts=${trust_hosts} -f ${DOCKER_FILE} .
+
+create-host-workspace: ##
+	## Create shared volume on the host for sharing files with the container.
+	##
+	mkdir -p ${HOST_WORK_DIR}
+
+run: create-host-workspace
+run: ##
+	## Start a Pyro CPU docker instance, and run the command `cmd`.
+	## Args:
+	##   img: use image name given by `img`.
+	##   cmd: command invoked on running a docker instance.
+	##     default - bash
+	##
+	docker run --init -it --user ${USER} \
+	-v ${HOST_WORK_DIR}:${DOCKER_WORK_DIR} \
+	${IMG_NAME} ${cmd}
+
+run-gpu: create-host-workspace
+run-gpu: ##
+	## Start a Pyro GPU docker instance, and run the command `cmd`.
+	## Args:
+	##   img: use image name given by `img`.
+	##   cmd: command invoked on running a docker instance.
+	##     default - bash
+	##
+	docker run --init --gpus=all -it --user ${USER} \
+	-v ${HOST_WORK_DIR}:${DOCKER_WORK_DIR} \
+	${IMG_NAME} ${cmd}
+
+notebook: create-host-workspace
+notebook: ##
+	## Start a jupyter notebook on the Pyro CPU docker container.
+	## Args:
+	##   img: use image name given by `img`.
+	##
+	docker run --init -it -p 8888:8888 --user ${USER} \
+	-v ${HOST_WORK_DIR}:${DOCKER_WORK_DIR} \
+	${IMG_NAME}
+
+notebook-gpu: create-host-workspace
+notebook-gpu: ##
+	## Start a jupyter notebook on the Pyro GPU docker container.
+	## Args:
+	##   img: use image name given by `img`.
+	##
+	docker run --gpus=all --init -it -p 8888:8888 --user ${USER} \
+	-v ${HOST_WORK_DIR}:${DOCKER_WORK_DIR} \
+	${IMG_NAME}
+
+	notebook: create-host-workspace
+lab: ##
+	## Start jupyterlab on the Pyro CPU docker container.
+	## Args:
+	##   img: use image name given by `img`.
+	##
+	docker run --init -it -p 8888:8888 --user ${USER} \
+	-v ${HOST_WORK_DIR}:${DOCKER_WORK_DIR} \
+	${IMG_NAME} jupyter lab --port=8888 --no-browser --ip=0.0.0.0
+
+lab-gpu: create-host-workspace
+lab-gpu: ##
+	## Start jupyterlab on the Pyro GPU docker container.
+	## Args:
+	##   img: use image name given by `img`.
+	##
+	docker run --gpus=all --init -it -p 8888:8888 --user ${USER} \
+	-v ${HOST_WORK_DIR}:${DOCKER_WORK_DIR} \
+	${IMG_NAME} jupyter lab --port=8888 --no-browser --ip=0.0.0.0
+
+

pyro/source/docker/README.md

@@ -0,0 +1,66 @@
+<!--
+Copyright Contributors to the Pyro project.
+
+SPDX-License-Identifier: Apache-2.0
+-->
+
+## Using Pyro Docker
+
+Some utilities for building docker images and running Pyro inside a Docker container are
+included in the `docker` directory. This includes a Dockerfile to build PyTorch and Pyro,
+with some common recipes included in the Makefile.
+ 
+Dependencies for building the docker images:
+ - **docker** (>= version 17.05)
+ - **nvidia-docker** Refer to the [readme](https://github.com/NVIDIA/nvidia-docker) for
+   installation.
+ 
+ 
+### Building Images
+
+The Makefile can be used to build CPU and CUDA images for Pyro and PyTorch. Some common
+options are as follows:
+
+ 1. **Source:** Uses the latest released package (conda package for PyTorch and PyPi wheel 
+    for Pyro) by default. However, both Pyro and PyTorch can be built from source from the
+    master branch or any other arbitrary branch specified by `pytorch_branch` and 
+    `pyro_branch`.
+ 2. **CPU / CUDA:** `make build` or `make build-gpu` can be used to specify whether the CPU
+    or the CUDA image is to be built. For building the CUDA image, *nvidia-docker* is 
+    required. 
+ 3. **Python Version:** Python version can be specified via the argument `python_version`. 
+ 
+For example, the `make` command to build an image that uses Pyro's `dev` branch over
+PyTorch's `master` branch, using python 3.6 to run on a GPU, is as follows:
+
+```sh
+make build-gpu pyro_branch=dev pytorch_branch=master python_version=3.6
+```  
+
+This will build an image named `pyro-gpu-dev-3.6`. To spin up a docker container from this
+image, and run jupyter notebook on this, use the following `make` command:
+
+```sh
+make notebook-gpu img=pyro-gpu-dev-3.6
+```
+
+For help on the `make` commands available, run `make help`.
+
+**NOTE (Mac Users)**: Please increase the memory available to the Docker application
+via *Preferences --> Advanced* from 2GB (default) to at least 4GB prior to building the
+docker image (specially for building PyTorch from source).
+
+### Running the Docker container
+
+Once the image is built, the docker container can be started via `make run`, or 
+`make run-gpu`. By default this starts a *bash* shell. One could start an *ipython* 
+shell instead by running `make run cmd=ipython`. The image to be used can be 
+specified via the argument `img`. 
+
+To run a *jupyter notebook* use `make notebook`, or `make notebook-gpu`. This will 
+start a jupyter notebook server which can be accessed from the browser using the link 
+mentioned in the terminal. 
+
+Note that there is a shared volume between the container and the host system, with the 
+location `$DOCKER_WORK_DIR` on the container, and `$HOST_WORK_DIR` on the local system.
+These variables can be configured in the `Makefile`.

pyro/source/docker/install.sh

@@ -0,0 +1,43 @@
+#!/usr/bin/env bash
+
+# Copyright Contributors to the Pyro project.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+set -xe
+
+pip install --upgrade pip
+pip install notebook ipywidgets matplotlib
+
+# 1. Install PyTorch
+# Use conda package if pytorch_branch = 'release'.
+# Else, install from source, using git branch `pytorch_branch`
+
+pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/${pytorch_whl}
+if [ ${pytorch_branch} != "release" ]
+then
+    git clone --recursive https://github.com/pytorch/pytorch.git
+    pushd pytorch && git checkout ${pytorch_branch}
+    pip uninstall -y torch
+    conda install cmake ninja
+    pip install -r requirements.txt
+    pip install mkl-static mkl-include
+    if [ ${pytorch_whl} != "cpu" ]
+    then
+        conda install -c pytorch magma-cuda${pytorch_whl:2}
+    fi
+    pip install -e .
+    popd
+fi
+
+
+# 2. Install Pyro
+# Use pypi wheel if pyro_branch = 'release'.
+# Else, install from source, using git branch `pyro_branch`
+if [ ${pyro_branch} = "release" ]
+then
+    pip install pyro-ppl
+else
+    git clone ${pyro_git_url}
+    (cd pyro && git checkout ${pyro_branch} && pip install -e .[dev])
+fi

pyro/source/docs/Makefile

@@ -0,0 +1,28 @@
+# Copyright Contributors to the Pyro project.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+# Minimal makefile for Sphinx documentation
+#
+
+# You can set these variables from the command line.
+SPHINXOPTS    ?= -E -W
+SPHINXBUILD   = python -msphinx
+APIDOC        = sphinx-apidoc
+SPHINXPROJ    = Pyro
+SOURCEDIR     = source
+PROJECTDIR    = ../pyro
+BUILDDIR      = build
+
+# Put it first so that "make" without argument is like "make help".
+help:
+	@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
+
+.PHONY: help Makefile
+
+apidoc:
+	$(APIDOC) -o "$(SOURCEDIR)" "$(PROJECTDIR)"
+# Catch-all target: route all unknown targets to Sphinx using the new
+# "make mode" option.  $(O) is meant as a shortcut for $(SPHINXOPTS).
+%: Makefile
+	@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)

pyro/source/docs/README.md

@@ -0,0 +1,26 @@
+<!--
+Copyright Contributors to the Pyro project.
+
+SPDX-License-Identifier: Apache-2.0
+-->
+
+# Documentation #
+Pyro Documentation is supported by [Sphinx](http://www.sphinx-doc.org/en/stable/). 
+To build the docs, run from the toplevel directory:
+```
+make docs
+```
+
+## Installation ##
+```
+pip install -r requirements.txt
+```
+
+Note that you will need to install [graphviz](https://www.graphviz.org/) separately.
+
+## Workflow ##
+To change the documentation, update the `*.rst` files in `source`.
+
+To build the docstrings, `sphinx-apidoc [options] -o <output_path> <module_path> [exclude_pattern, ...]`
+
+To build the html pages, `make html`

pyro/source/docs/requirements.txt

@@ -0,0 +1,15 @@
+# Copyright Contributors to the Pyro project.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+sphinx==4.2.0
+sphinx-rtd-theme==1.0.0
+graphviz>=0.8
+numpy>=1.7
+observations>=0.1.4
+opt_einsum>=2.3.2
+pyro-api>=0.1.1
+tqdm>=4.36
+funsor[torch]
+setuptools
+sphinx_copybutton

pyro/source/docs/source/_static/css/pyro.css

@@ -0,0 +1,35 @@
+/*
+ * Copyright Contributors to the Pyro project.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+@import url("theme.css");
+
+.wy-side-nav-search {
+    background-color: #565656;
+}
+
+.wy-side-nav-search a {
+    margin: 0
+}
+
+.wy-side-nav-search > div.version {
+    color: #f26822;
+}
+
+.wy-nav-top {
+    background: #404040;
+}
+
+.wy-menu-vertical li.on a, .wy-menu-vertical li.current>a {
+    background: #ccc;
+}
+
+.wy-side-nav-search input[type=text] {
+    border-color: #313131;
+}
+
+.wy-side-nav-search>a img.logo, .wy-side-nav-search .wy-dropdown>a img.logo {
+    max-width: 60%;
+}