View source on GitHub
|
Module for probabilistic programming features.
Modules
effect_handler module: Enables writing custom effect handlers for probabilistic programs.
plate_util module: Contains utilities for the plate transformation.
transformations module: Module for probabilistic programming transformations.
Functions
block(...): Returns a program that removes the provided names from transformations.
conditional(...): Conditions a probabilistic program on random variables.
graph_replace(...): Transforms a program to one with new inputs and outputs.
intervene(...): Transforms a program into one where provided random variables are fixed.
joint_log_prob(...): Returns a function that computes the log probability of all of a program's random variables.
joint_sample(...): Returns a program that outputs a dictionary of latent random variable samples.
log_prob(...): Returns a function that computes the log probability of a sample.
make_effect_handler(...): Returns a function transformation that applies a provided set of handlers.
nest(...): Wraps a function to create a new scope for harvested values.
plate(...): Transforms a program into one that draws samples on a named axis.
random_variable(...): A single-dispatch function used to tag values and the outputs of programs.
rv(...): A single-dispatch function used to tag values and the outputs of programs.
trace(...): Returns a program that additionally outputs sampled latents.
trace_log_prob(...): Returns a function that computes the log probability program's output and its random variables.
Type Aliases
Other Members | |
|---|---|
| RANDOM_VARIABLE |
'random_variable'
|