Program transforms

Program transforms.

detach(p)[source]

Makes random variables in p become non-reparameterized.

empirical(out, trace, metrics)[source]

Creates an empirical program given a trace.

prng_key()[source]

Generates a random JAX PRNGKey.

register_backend(backend, traced_evaluate=None, empirical=None, suffix=None, prng_key=None, detach=None, stick_the_landing=None)[source]

Register backend.

set_backend(backend)[source]

Set backend.

stick_the_landing(p)[source]

Stops gradient of distributions’ parameters before computing log prob.

suffix(p)[source]

Adds suffix _PREV_ to variable names of p.

traced_evaluate(p, latents=None, seed=None, **kwargs)[source]

Performs traced evaluation for a program p.