numpyro

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

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)

Python library for Multi-Armed Bandits

Bayesian Ecological Modeling in Python

The missing layers package for Bayesian regression

Estimating time trees from very large phylogenies

Efficient library for spectral analysis in high-energy astrophysics.

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

Scalable Bayesian Modelling: A comparison

My implementation of John K. Kruschke's Doing Bayesian Data Analysis 2nd edition using Python and Numpyro.

Hierarchical Bayesian estimation of MEP recruitment curves

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

Tutorials for the 2022 IAIFI Summer School, covering (deep) probabilistic programming with Jax and NumPyro.

Build, fit, and sample from cognitive models with JAX + NumPyro.

Bridging simulators to probabilistic programming in JAX

Mixture regression models for NumPyro.

Do LLM judges know when they're wrong? Three open-weight judges (Qwen2.5-7B, kev-8b, auto-j-13b) graded against MT-Bench human votes: calibration, position and padding attacks, and what auto-accepting confident verdicts would cost. Interactive site included.