GenJAX API Reference

JAX-based probabilistic programming with programmable inference.

Installation

pip install genjax

Modules

• genjax.core - Core functionality and Generative Function Interface
• genjax.distributions - Probability distributions
• genjax.inference.mcmc - Markov Chain Monte Carlo
• genjax.inference.smc - Sequential Monte Carlo
• genjax.pjax - Probabilistic JAX primitives
• genjax.adev - Automatic differentiation of expected values
• genjax.state - State interpreter
• genjax.sp - Structural primitives