def METHOD_NAME(new_states: Iterable[Tensor]) -> None:
    r"""Sets the random number generator state of all devices.
    Args:
        new_states (Iterable of torch.ByteTensor): The desired state for each device"""
    for i, state in enumerate(new_states):
        set_rng_state(state, i)