API Reference¶
Complete API documentation for GenJAX, automatically generated from source code.
Core Modules¶
genjax.core¶
Core functionality including the Generative Function Interface (GFI), traces, and the @gen decorator.
genjax.distributions¶
Built-in probability distributions that implement the GFI.
genjax.pjax¶
Probabilistic JAX (PJAX) - foundational probabilistic programming primitives.
Inference¶
genjax.inference.mcmc¶
Markov Chain Monte Carlo algorithms including Metropolis-Hastings and HMC.
genjax.inference.smc¶
Sequential Monte Carlo methods for particle-based inference.
genjax.inference.vi¶
Variational inference algorithms and gradient estimators.
Advanced¶
genjax.adev¶
Automatic differentiation of expected values for gradient estimation.
genjax.state¶
State interpreter for inspecting and manipulating probabilistic computations.
Utilities¶
genjax.sp¶
Structural primitives and combinators for building complex models.
genjax.timing¶
Utilities for benchmarking and performance analysis.