Save the date! Google I/O returns May 18-20 Register now


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.

The input value.