def reset(self):
        """Set to the initial state (no tracker)."""
        self.states_ = {}
        self.seeds_ = set()
Set to the initial state (no tracker).
dpkd/transformers/src/transformers/mpu/random.py