def METHOD_NAME(self) -> List[Parameter]:
    if self.offloader is None:
        return
    assert len(self.prefetch_params) == 0 and len(self.param_to_prefetch_idx) == 0
    for group in self.param_groups:
        for p in group['params']:
            if p.grad is None:
                continue
            if len(self.state[p]) > 0 and self.is_on_nvme[p]:
                assert p.device.type == 'cpu'
                self.param_to_prefetch_idx[p] = len(self.prefetch_params)
                self.prefetch_params.append(p)