Research index
GenJAX: Probabilistic Programming with Vectorized Programmable Inference
Becker, Huot, Matheos and 7 others
Zenodo (CERN European Organization for Nuclear Research)13 Oct 2025Open access
NobleIDark:/48914/w1/20260523/F7C7FDD5DOI10.5281/zenodo.17344136
GenJAX: Probabilistic Programming with Vectorized Programmable Inference
Becker, Huot, Matheos and 7 others · Zenodo (CERN European Organization for Nuclear Research) · 2025
NobleIDark:/48914/w1/20260523/F7C7FDD5DOI10.5281/zenodo.17344136
Free to read at the publisher
You can read the abstract here, or the full text at the publisher.
Abstract
GenJAX is a probabilistic programming language (PPL) designed around programmable inference - automation that allows users to express and customize Bayesian inference algorithms. The system provides modeling language automation for constructing complex probability distributions and inference automation for constructing Monte Carlo samplers and variational inference methods. This repository includes the GenJAX implementation, documentation, and case studies from the POPL 2026 paper 'Probabilistic Programming with Vectorized Programmable Inference'.