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

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

Statistical Rethinking (2nd ed.) with NumPyro

Statistical Rethinking: A Bayesian Course Using Python and NumPyro

Fully and Partially Bayesian Neural Nets

A multiverse of Prophet models for timeseries

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

Bayesian Learning and Neural Networks (jupyter book sources)

The missing layers package for Bayesian inference

Bayesian Ecological Modeling in Python

Compares Stan, PyMC, and PyMC + JAX numpyro sampler on a model for tennis

A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work

Estimating time trees from very large phylogenies

Python package for multi-signal Bayesian renewal modeling with JAX and NumPyro.

GWInferno: Gravitational-Wave Hierarchical Inference with NumPyro

Efficient library for spectral analysis in high-energy astrophysics.

An extension of NumPyro for first-class support of dynamical systems.

No description available

A numpyro implementation of BayeSN, the optical-NIR SED model for type Ia supernovae

An implementation of the differentially private variational inference algorithm for NumPyro.

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

Probabilistic deep learning using JAX

Numpyro examples in Python notebooks

A wrapper API for working with numpyro models.

A numpyro Gibbs sampler that uses conditioned HMC kernels for each step.

Scalable Bayesian Modelling: A comparison

jax, numpyro samples

No description available

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.