Module: oryx.core.ppl

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

LogProbFunction

Program

RANDOM_VARIABLE 'random_variable'