oryx.core.ppl.random_variable

View source on GitHub

A single-dispatch function used to tag values and the outputs of programs.

View aliases

Main aliases

oryx.core.ppl.rv, oryx.core.ppl.transformations.random_variable, oryx.core.ppl.transformations.rv

oryx.core.ppl.random_variable(
    obj, *, name: Optional[str] = None, plate: Optional[str] = None
) -> oryx.core.ppl.LogProbFunction

Used in the notebooks

Used in the tutorials
Probabilistic Programming in Oryx

random_variable is a single-dispatch function that enables registering custom types. Its default implementation is to tag input value with a name and return it.

Args
obj	A JAX type to be tagged.
name	A string name to tag input value, cannot be None.
plate	A string named axis for this random variable's plate.

Returns
The input value.