'Set the RNG state for both the on device and on host RNGs.
Arguments:
rng_states (tuple of np.arrays): tuple with 2 elements
1) numpy random number state vector
2) array of uint32 specifying on dev RNG state'

def rng_set_state(self, rng_states):

    assert ((type(rng_states) is tuple) and (len(rng_states) == 2))
    self._set_rand_state_dev(state=rng_states[1])
    self.rng.set_state(rng_states[0])
