Method prepare

src/fused_mm_sampling/core.py:645–646

(self)

class Sampler(Protocol):
    def prepare(self) -> "Sampler":
        raise NotImplementedError()

    def sample(self, **kwargs) -> torch.Tensor:
        raise NotImplementedError()