def METHOD_NAME(self, callbacks):
    """Register callbacks that will be invoked following a reset event (worker added or removed).
    For example, a common use of a reset callback would be to update the learning rate scale with the
    new number of workers.
    Args:
        callbacks: list of functions to execute.
    """
    self._reset_callbacks.extend(callbacks)