There are 2 repositories under numpyro topic.
Fully and Partially Bayesian Neural Nets
Course materials of "Bayesian Modelling and Probabilistic Programming with Numpyro, and Deep Generative Surrogates for Epidemiology"
A multiverse of Prophet models for timeseries
Estimating time trees from very large phylogenies
The missing layers package for Bayesian inference
JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro
My implementation of John K. Kruschke's Doing Bayesian Data Analysis 2nd edition using Python and Numpyro.
Tutorials for the 2022 IAIFI Summer School, covering (deep) probabilistic programming with Jax and NumPyro.
Summary notebooks using derivative gaussian processes with tinygp. We implement a 2D derivative gaussian process and successfully use derivatives to regularize SVI fits with a gaussian process model..
Very easy Bayesian regression.
Bayesian inference using sparse gaussian processes from tinygp. Examples include 1D and 2D implementation.
Oxford MSc thesis. variational autoencoder combined with graph convolutional networks for learning locally-aware spatial prior distributions
Statistical rethinking by Richard McElreath. Learning notes, code port to PyMC (mainly for MCMC) v5 & Numpyro (mainly for `quap`).
Repo for course CSC2558: "Intelligent Adaptive Interventions" project in nonstationary contextual bandits.
Summary notebook implementing Bayesian Model Averaging with numpyro.
Novel technique to fit a target distribution with a class of distributions using SVI (via NumPyro). Unlike standard SVI, our "data" is a distribution rather than a finite collection of samples.
Pretty, easy, flexible Bayesian estimation with data overlay
Measuring how events shape discourse on twitter surrounding the Ukraine War in 2022 using piecewise exponential decay models.
[Break Through Tech: AI Studio Project] Unwrapping Customer Delight: Using Frequentist and Bayesian Regression Models to Optimize Surprise Gift Strategies
Parameter estimation for the Compound Poisson Gamma (Tweedie) distribution in NumPyro
A basic example to run ML code on SLURM (specifically Oxford's ARC systems) in a organised fashion
Bayesian Analysis in Python (2nd ed.) with Numpyro