Getting Started | Documentation | Community | Contributing
Pyro is a flexible, scalable deep probabilistic programming library built on PyTorch. Notably, it was designed with these principles in mind:
Pyro is in a beta release. It is developed and maintained by Uber AI Labs and community contributors. For more information, check out our blog post.
Install using pip:
Pyro supports Python 3.4+.
pip install pyro-ppl
Install from source:
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:
pip install pyro-ppl[extras]
Make sure that the models come from the same release version of the Pyro source code as you have installed.
For recent features you can install Pyro from source.
Install using pip:
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:
pip install git+https://github.com/pyro-ppl/pyro.git#egg=project[extras]
Install from source:
git clone https://github.com/pyro-ppl/pyro
cd pyro
pip install . # pip install .[extras] for running models in examples/tutorials
Refer to the instructions here.
If you use Pyro, please consider citing:
@article{bingham2018pyro,
author = {Bingham, Eli and Chen, Jonathan P. and Jankowiak, Martin and Obermeyer, Fritz and
Pradhan, Neeraj and Karaletsos, Theofanis and Singh, Rohit and Szerlip, Paul and
Horsfall, Paul and Goodman, Noah D.},
title = {{Pyro: Deep Universal Probabilistic Programming}},
journal = {arXiv preprint arXiv:1810.09538},
year = {2018}
}