ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


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 (str): A string name to tag input value, cannot be None. plate (str): A string named axis for this random variable's plate.

The input value.