numpyro

Python code for "Probabilistic Machine learning" book by Kevin Murphy

Fully and Partially Bayesian Neural Nets

Course materials of "Bayesian Modelling and Probabilistic Programming with Numpyro, and Deep Generative Surrogates for Epidemiology"

Bayesian Learning and Neural Networks (jupyter book sources)

A multiverse of Prophet models for timeseries

Estimating time trees from very large phylogenies

Efficient library for spectral analysis in high-energy astrophysics.

Probabilistic deep learning using JAX

JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro

Bayesian Ecological Modeling in Python

The missing layers package for Bayesian inference

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.

Scalable Bayesian Modelling: A comparison

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.

Oxford MSc thesis. variational autoencoder combined with graph convolutional networks for learning locally-aware spatial prior distributions

Mixture regression models for NumPyro.

Bayesian inference using sparse gaussian processes from tinygp. Examples include 1D and 2D implementation.

Repo for course CSC2558: "Intelligent Adaptive Interventions" project in nonstationary contextual bandits.