Code Examples

Examples on specifying models and doing inference in NumPyro.

View source files on github

Example: Conditional Variational Autoencoder in Flax

Example: Conditional Variational Autoencoder in Flax

Example: Neal’s Funnel

Example: Neal's Funnel

Example: Stochastic Volatility

Example: Stochastic Volatility

Example: Predator-Prey Model

Example: Predator-Prey Model

Example: Nested Sampling for Gaussian Shells

Example: Nested Sampling for Gaussian Shells

Example: AR2 process

Example: AR2 process

Example: Generalized Linear Mixed Models

Example: Generalized Linear Mixed Models

Example: Toy Mixture Model with Discrete Enumeration

Example: Toy Mixture Model with Discrete Enumeration

Example: AutoDAIS

Example: AutoDAIS

Example: Proportion Test

Example: Proportion Test

Example: Modelling mortality over space and time

Example: Modelling mortality over space and time

Example: Zero-Inflated Poisson regression model

Example: Zero-Inflated Poisson regression model

Example: Hamiltonian Monte Carlo with Energy Conserving Subsampling

Example: Hamiltonian Monte Carlo with Energy Conserving Subsampling

Example: Holt-Winters Exponential Smoothing

Example: Holt-Winters Exponential Smoothing

Example: Bayesian Neural Network

Example: Bayesian Neural Network

Example: Horseshoe Regression

Example: Horseshoe Regression

Example: VAR(2) process

Example: VAR(2) process

Example: Variational Autoencoder

Example: Variational Autoencoder

Example: Bayesian Neural Network with SteinVI

Example: Bayesian Neural Network with SteinVI

Example: Neural Transport

Example: Neural Transport

Example: Gaussian Process

Example: Gaussian Process

Example: Baseball Batting Average

Example: Baseball Batting Average

Example: MCMC Methods for Tall Data

Example: MCMC Methods for Tall Data

Example: Hidden Markov Model

Example: Hidden Markov Model

Example: Thompson sampling for Bayesian Optimization with GPs

Example: Thompson sampling for Bayesian Optimization with GPs

Example: Sine-skewed sine (bivariate von Mises) mixture

Example: Sine-skewed sine (bivariate von Mises) mixture

Example: ProdLDA with Flax and Haiku

Example: ProdLDA with Flax and Haiku

Example: Deep Markov Model inferred using SteinVI

Example: Deep Markov Model inferred using SteinVI

Example: CJS Capture-Recapture Model for Ecological Data

Example: CJS Capture-Recapture Model for Ecological Data

Example: Bayesian Models of Annotation

Example: Bayesian Models of Annotation

Example: Enumerate Hidden Markov Model

Example: Enumerate Hidden Markov Model

Example: Sparse Regression

Example: Sparse Regression

Example: Hilbert space approximation for Gaussian processes.

Example: Hilbert space approximation for Gaussian processes.