API and Developer Reference
detach()
empirical()
prng_key()
register_backend()
set_backend()
stick_the_landing()
suffix()
traced_evaluate()
Tutorials and Examples
Program transforms.
Makes random variables in p become non-reparameterized.
Creates an empirical program given a trace.
Generates a random JAX PRNGKey.
Register backend.
Set backend.
Stops gradient of distributions’ parameters before computing log prob.
Adds suffix _PREV_ to variable names of p.
Performs traced evaluation for a program p.