There are 2 repositories under numpyro topic.
A multiverse of Prophet models for timeseries
Estimating time trees from very large phylogenies
Probabilistic deep learning using JAX
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. PriorVAE with graph convolutional networks for learning locally-aware spatial prior distributions
Repo for course CSC2558: "Intelligent Adaptive Interventions" project in nonstationary contextual bandits.
Statistical rethinking by Richard McElreath. Learning notes, code port to PyMC (mainly for MCMC) v5 & Numpyro (mainly for `quap`).
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.
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