def jitter_augment(is_training=True, **kwargs):
  """Applies random color jitter augmentation."""
  if is_training:
    brightness = kwargs['brightness'] if 'brightness' in kwargs else 0.125
    contrast = kwargs['contrast'] if 'contrast' in kwargs else 0.4
    saturation = kwargs['saturation'] if 'saturation' in kwargs else 0.4
    hue = kwargs['hue'] if 'hue' in kwargs else 0
    return [('jitter', {
        'brightness': brightness,
        'contrast': contrast,
        'saturation': saturation,
        'hue': hue
    })]
  return []