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

Used in the notebooks

Used in the tutorials

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.

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.

The input value.