def METHOD_NAME(
    self, lazy_awaitable_kt: LazyAwaitable[KeyedTensor]
) -> LazyAwaitable[KeyedTensor]:
    # This will ensure that all feature processor parameters participate in the
    # autograd graph across all ranks. This will protect from mismatched collective
    # calls order when using DistributedDataParallel over feature processors.
    no_op_tensor = (
        self._no_op_zero
        * torch.cat(
            [x.flatten() for x in self._feature_processors.parameters()]
        ).sum()
    )
    lazy_awaitable_kt.callbacks.append(
        partial(param_dp_sync, no_op_tensor=no_op_tensor)
    )
    return lazy_awaitable_kt