NumPyro documentation
API and Developer Reference
Introductory Tutorials

Bayesian Regression Using NumPyro

Bayesian Hierarchical Linear Regression

Example: Baseball Batting Average

Example: Variational Autoencoder

Example: Neal’s Funnel

Example: Stochastic Volatility

Example: ProdLDA with Flax and Haiku

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
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