Method sample

src/fused_mm_sampling/core.py:659–660

(self, **kwargs)

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

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