Getting Started with NumPyro

API and Developer Reference

Pyro Primitives
Distributions
Inference
Effect Handlers
Contributed Code
Change Log

Introductory Tutorials

Bayesian Regression Using NumPyro
Bayesian Hierarchical Linear Regression
Example: Baseball Batting Average
Effect Handlers in NumPyro: A Practical Guide
Example: Variational Autoencoder
Example: Neal’s Funnel
Example: Stochastic Volatility
Example: ProdLDA with Flax
Variationally Inferred Parameterization
Automatic rendering of NumPyro models
Bad posterior geometry and how to deal with it
Truncated and folded distributions
Bayesian Censoring Data Modeling
Hilbert Space Approximation Gaussian Process Module
NumPyro Integration with Other Libraries
Fast Gaussian Process Inference Using Circulant Normal Distributions
NNX and NumPyro Integration
Composing Gaussian Processes with NumPyro using GPJax
Embarrassingly Parallel MCMC via Consensus Monte Carlo

Discrete Latent Variables

Gaussian Mixture Model
Example: Toy Mixture Model with Discrete Enumeration
Example: Bayesian Models of Annotation
Example: Enumerate Hidden Markov Model
Example: CJS Capture-Recapture Model for Ecological Data
Example: Nested Sampling for Gaussian Shells
Bayesian Imputation for Missing Values in Discrete Covariates

Applications

Time Series Forecasting
Ordinal Regression
Bayesian Imputation
Example: Gaussian Process
Example: Bayesian Neural Network
Example: AutoDAIS
Example: Sparse Regression
Example: Horseshoe Regression
Example: Proportion Test
Example: Generalized Linear Mixed Models
Example: Hidden Markov Model
Example: Hilbert space approximation for Gaussian processes.
Example: Hilbert space approximation for Gaussian processes (multidimensional)
Example: Predator-Prey Model
Solving differential equations (ODEs) for multiple initial conditions.
Example: Neural Transport
Example: Thompson sampling for Bayesian Optimization with GPs
Bayesian Hierarchical Stacking: Well Switching Case Study
Example: Sine-skewed sine (bivariate von Mises) mixture
Example: AR2 process
Example: Holt-Winters Exponential Smoothing
Example: Modelling mortality over space and time
Example: Zero-Inflated Poisson regression model
Example: Conditional Variational Autoencoder in Flax
Text-Based Ideal Points using NumPyro
Example: VAR(2) process
Hierarchical Forecasting
Bayesian CUPED

Other Inference Algorithms

Example: MCMC Methods for Tall Data
Example: Hamiltonian Monte Carlo with Energy Conserving Subsampling
Example: Bayesian Neural Network with SteinVI
Example: Deep Markov Model inferred using SteinVI

NumPyro

Overview: module code

All modules for which code is available

numpyro.contrib.control_flow.cond
numpyro.contrib.control_flow.scan
numpyro.contrib.ecs_proxies
numpyro.contrib.einstein.stein_kernels
numpyro.contrib.einstein.steinvi
numpyro.contrib.funsor.discrete
numpyro.contrib.funsor.enum_messenger
numpyro.contrib.funsor.infer_util
numpyro.contrib.hsgp.approximation
numpyro.contrib.hsgp.laplacian
numpyro.contrib.hsgp.spectral_densities
numpyro.contrib.module
numpyro.contrib.nested_sampling
numpyro.contrib.stochastic_support.dcc
numpyro.contrib.stochastic_support.sdvi
numpyro.contrib.tfp.distributions
numpyro.contrib.tfp.mcmc
numpyro.diagnostics
numpyro.distributions.censored
numpyro.distributions.conjugate
numpyro.distributions.constraints
numpyro.distributions.continuous
numpyro.distributions.copula
numpyro.distributions.directional
numpyro.distributions.discrete
numpyro.distributions.distribution
numpyro.distributions.flows
numpyro.distributions.mixtures
numpyro.distributions.transforms
numpyro.distributions.truncated
numpyro.distributions.util
numpyro.handlers
numpyro.infer.autoguide
numpyro.infer.barker
numpyro.infer.elbo
numpyro.infer.ensemble
numpyro.infer.hmc
numpyro.infer.hmc_gibbs
numpyro.infer.hmc_util
numpyro.infer.importance
numpyro.infer.initialization
numpyro.infer.inspect
numpyro.infer.mcmc
numpyro.infer.mixed_hmc
numpyro.infer.reparam
numpyro.infer.sa
numpyro.infer.svi
numpyro.infer.util
numpyro.ops.indexing
numpyro.optim
numpyro.primitives
numpyro.util

© Copyright 2019, Uber Technologies, Inc.

Built with Sphinx using a theme provided by Read the Docs.