numpyro

Fully and Partially Bayesian Neural Nets

A multiverse of Prophet models for timeseries

Python library for Multi-Armed Bandits

The missing layers package for Bayesian regression

Estimating time trees from very large phylogenies

Efficient library for spectral analysis in high-energy astrophysics.

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

Hierarchical Bayesian estimation of MEP recruitment curves

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

Bridging simulators to probabilistic programming in JAX

Very easy Bayesian regression.

Mixture regression models for NumPyro.

Hierarchical Bayesian prediction for bounded scores of events nested in entities over time — configured by one YAML descriptor.

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

This project is not well tested, or thoroughly vetted. Use with extreme care.

Codebase for my submission to Kaggle's yearly March Machine Learning Mania competition

Physics-Informed ML for population-level inverse problems | Cambridge MEng Thesis (First Class)

Bayesian JWST secondary-eclipse depth inference for rocky exoplanets — a reproducible JAX/NumPyro pipeline (Rocky Worlds DDT Data Challenge)

Engine for validating ranked security taxonomies against incident corpora via Bayesian inference. First cycle: OWASP LLM Top 10 2026.

Bayesian prioritization framework for Near-Earth Object follow-up observations in the Rubin/LSST era. Research-grade follow-up to neo-triage-hack hackathon submission.