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


Named tuple containing both original_inputs and min_diff_data.

MinDiffModel default implementations and utils.(un)pack_* functions use this class to pack and unpack the separate components required for MinDiff and regular training.

original_inputs Batch of inputs that would originally (i.e. without applying MinDiff) be passed in to a model's method. This corresponds to the x component described in
min_diff_data Batch of supplemental data to be used to calculate the min_diff_loss.