numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
File Explorer
Download Latest Version (.zip)- bug_report.md
- config.yml
- feature_request.md
- post-sticky-comment.js
- benchmark-comment.yml
- benchmark.yml
- ci.yml
- pr-cleanup.yml
- pr-triage.yml
- publish.yml
- __init__.py
- distributions.py
- handlers.py
- mcmc.py
- svi.py
- __init__.py
- compare.py
- harness.py
- models.py
- README.md
- runner.py
- Dockerfile
- Dockerfile
- README.md
- pyro.css
- annotation.png
- ar2.png
- baseball.png
- bnn.png
- capture_recapture.png
- covtype.png
- cvae.png
- funnel.png
- gaussian_shells.png
- gp.png
- hmcecs.png
- hmm.png
- hmm_enum.png
- holt_winters.png
- hsgp.png
- mortality.png
- neutra.png
- ode.png
- prodlda.png
- proportion_test.png
- sparse_regression.png
- ssbvm_mixture.png
- ssbvm_mixture_torus_top.png
- stein_bnn.png
- stein_dmm.png
- stochastic_volatility.png
- thompson_sampling.png
- ucbadmit.png
- vae.png
- var2.png
- zero_inflated_poisson.png
- android-icon-144x144.png
- android-icon-192x192.png
- android-icon-36x36.png
- android-icon-48x48.png
- android-icon-72x72.png
- android-icon-96x96.png
- apple-icon-114x114.png
- apple-icon-120x120.png
- apple-icon-144x144.png
- apple-icon-152x152.png
- apple-icon-180x180.png
- apple-icon-57x57.png
- apple-icon-60x60.png
- apple-icon-72x72.png
- apple-icon-76x76.png
- apple-icon-precomposed.png
- apple-icon.png
- browserconfig.xml
- favicon-16x16.png
- favicon-32x32.png
- favicon-96x96.png
- favicon.ico
- manifest.json
- ms-icon-144x144.png
- ms-icon-150x150.png
- ms-icon-310x310.png
- ms-icon-70x70.png
- bayesian_cuped.png
- bayesian_hierarchical_linear_regression.png
- bayesian_hierarchical_stacking.png
- bayesian_imputation.png
- bayesian_regression.png
- censoring.png
- circulant_gp.png
- consensus_mc.png
- discrete_imputation.png
- effect_handlers.png
- gpjax_example.png
- hierarchical_forecasting.png
- hsgp_example.png
- hsgp_nd_example.png
- model_rendering.png
- nnx_example.png
- ordinal_regression.png
- other_samplers.png
- tbip.png
- time_series_forecasting.png
- variationally_inferred_parameterisation.png
- dais_demo.png
- pyro_logo.png
- pyro_logo_small.png
- pyro_logo_wide.png
- pyro_logo_with_text.png
- breadcrumbs.html
- autoguide.rst
- conf.py
- contrib.rst
- diagnostics.rst
- distributions.rst
- funsor.rst
- handlers.rst
- index.rst
- infer.rst
- mcmc.rst
- optimizers.rst
- primitives.rst
- reparam.rst
- svi.rst
- utilities.rst
- Makefile
- README.md
- __init__.py
- data.py
- main.py
- models.py
- README.md
- train_baseline.py
- train_cvae.py
- __init__.py
- annotation.py
- ar2.py
- baseball.py
- bnn.py
- capture_recapture.py
- covtype.py
- cvae.py
- dais_demo.py
- funnel.py
- gaussian_shells.py
- gp.py
- hmcecs.py
- hmm.py
- hmm_enum.py
- holt_winters.py
- horseshoe_regression.py
- hsgp.py
- minipyro.py
- mortality.py
- neutra.py
- ode.py
- prodlda.py
- proportion_test.py
- README.rst
- sparse_regression.py
- ssbvm_mixture.py
- stein_bnn.py
- stein_dmm.py
- stochastic_volatility.py
- thompson_sampling.py
- toy_mixture_model_discrete_enumeration.py
- ucbadmit.py
- vae.py
- var2.py
- zero_inflated_poisson.py
- pyro.css
- pyro_logo_wide.png
- bad_posterior_geometry.ipynb
- bayesian_cuped.ipynb
- bayesian_hierarchical_linear_regression.ipynb
- bayesian_hierarchical_stacking.ipynb
- bayesian_imputation.ipynb
- bayesian_regression.ipynb
- censoring.ipynb
- circulant_gp.ipynb
- conf.py
- consensus_mc.ipynb
- discrete_imputation.ipynb
- effect_handlers.ipynb
- gmm.ipynb
- gpjax_example.ipynb
- hierarchical_forecasting.ipynb
- hsgp_example.ipynb
- hsgp_nd_example.ipynb
- index.rst
- logistic_regression.ipynb
- lotka_volterra_multiple.ipynb
- model_rendering.ipynb
- nnx_example.ipynb
- ordinal_regression.ipynb
- other_samplers.ipynb
- tbip.ipynb
- time_series_forecasting.ipynb
- truncated_distributions.ipynb
- variationally_inferred_parameterization.ipynb
- Makefile
- __init__.py
- distributions.py
- handlers.py
- infer.py
- ops.py
- optim.py
- pyro.py
- util.py
- __init__.py
- cond.py
- scan.py
- __init__.py
- mixture_guide_predictive.py
- stein_kernels.py
- stein_loss.py
- stein_util.py
- steinvi.py
- __init__.py
- discrete.py
- enum_messenger.py
- infer_util.py
- __init__.py
- approximation.py
- laplacian.py
- spectral_densities.py
- __init__.py
- dcc.py
- sdvi.py
- __init__.py
- distributions.py
- mcmc.py
- __init__.py
- ecs_proxies.py
- module.py
- nested_sampling.py
- render.py
- __init__.py
- batch_util.py
- censored.py
- conjugate.py
- constraints.py
- continuous.py
- copula.py
- directional.py
- discrete.py
- distribution.py
- flows.py
- gof.py
- kl.py
- mixtures.py
- transforms.py
- truncated.py
- util.py
- __init__.py
- datasets.py
- __init__.py
- autoguide.py
- barker.py
- calibration.py
- elbo.py
- ensemble.py
- ensemble_util.py
- hmc.py
- hmc_gibbs.py
- hmc_util.py
- importance.py
- initialization.py
- inspect.py
- mcmc.py
- mixed_hmc.py
- reparam.py
- sa.py
- svi.py
- util.py
- __init__.py
- auto_reg_nn.py
- block_neural_arn.py
- masked_dense.py
- __init__.py
- indexing.py
- provenance.py
- pytree.py
- __init__.py
- _typing.py
- diagnostics.py
- handlers.py
- optim.py
- patch.py
- primitives.py
- util.py
- version.py
- update_headers.py
- update_version.py
- test_mixture_guide_predictive.py
- test_stein_kernels.py
- test_stein_loss.py
- test_steinvi.py
- test_steinvi_util.py
- test_approximation.py
- test_laplacian.py
- test_spectral_densities.py
- test_dcc.py
- test_sdvi.py
- test_control_flow.py
- test_enum_elbo.py
- test_esc_proxies.py
- test_funsor.py
- test_infer_discrete.py
- test_module.py
- test_nested_sampling.py
- test_tfp.py
- test_autoguide.py
- test_calibration.py
- test_ensemble_mcmc.py
- test_ensemble_util.py
- test_external_helpers.py
- test_gradient.py
- test_hmc_gibbs.py
- test_hmc_util.py
- test_importance.py
- test_infer_util.py
- test_inspect.py
- test_mcmc.py
- test_reparam.py
- test_svi.py
- test_indexing.py
- test_provenance.py
- conftest.py
- test_pyroapi.py
- conftest.py
- test_compile.py
- test_constraints.py
- test_diagnostics.py
- test_distributions.py
- test_distributions_mixture.py
- test_distributions_util.py
- test_example_utils.py
- test_examples.py
- test_flows.py
- test_gof.py
- test_handlers.py
- test_model_rendering.py
- test_nn.py
- test_optimizers.py
- test_pickle.py
- test_transforms.py
- test_util.py
- .gitattributes
- .gitignore
- .pre-commit-config.yaml
- .readthedocs.yaml
- CONTRIBUTING.md
- LICENSE.md
- Makefile
- pyproject.toml
- README.md
- tombi.toml
# Installation Guide
1. Get the code
git clone https://github.com/pyro-ppl/numpyro
Downloads the entire project code from GitHub to your computer.
cd numpyro
Moves into the project folder you just downloaded.
2. Official Install Script
Easy RecommendedPrerequisites
- Python 3 Python is required to use pip.
pip install numpyro
Installs the package published on PyPI directly β no need to clone the source.
pip install 'numpyro[cpu]'
Installs the package published on PyPI directly β no need to clone the source.
pip install 'numpyro[cuda12]' -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Installs the package published on PyPI directly β no need to clone the source.
pip install 'numpyro[cuda13]' -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Installs the package published on PyPI directly β no need to clone the source.
After installing, open a new terminal and run the program's version command (e.g. --version) to confirm it worked.
Pulled directly from this repo's README.
3. Docker
EasyPrerequisites
- Git Needed to download the project code from GitHub.
- Docker Desktop Needed to build and run containers. Install it and keep it running in the background.
β οΈ This is a large repository, so this method may point to an internal sub-package rather than the actual core product. Check the full README as well.
docker build -f docker/dev/Dockerfile -t numpyro .
Builds a runnable image based on the Dockerfile.
docker run -p 8080:80 numpyro
Runs the built image as an actual container.
Run docker compose ps to check the containers are Up. If the README mentions a port, open http://localhost:PORT in your browser.
4. Python
EasyPrerequisites
pip install numpyro
Installs the package published on PyPI directly β no need to clone the source.
pip install 'numpyro[cpu]'
Installs the package published on PyPI directly β no need to clone the source.
pip install 'numpyro[cuda12]' -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Installs the package published on PyPI directly β no need to clone the source.
pip install 'numpyro[cuda13]' -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Installs the package published on PyPI directly β no need to clone the source.
pip install -e '.[dev]' # contains additional dependencies for NumPyro development
Installs the Python libraries listed in requirements.txt (or similar).
If it runs without errors and prints output in the terminal, it worked.
Pulled directly from this repo's README.
5. Make
MediumPrerequisites
- Git Needed to download the project code from GitHub.
- Make Usually pre-installed on Linux/macOS. On Windows, install separately (e.g. via MSYS2 or WSL).
make
Compiles the code based on the generated build configuration to produce an executable.
If it finishes without errors, it worked. Try running the generated executable directly.
// repository documentation
Was this content helpful?
(0 ratings)