Method from_weights

src/fused_mm_sampling/core.py:670–677

(
        cls,
        weights: torch.Tensor,  # [V, D]
        epsilon: float = 0.2,
    )

    fn: Callable[..., torch.Tensor]

    def prepare(self) -> "SimpleSampler":
        return self

    def sample(self, **kwargs) -> torch.Tensor:
        return self.fn(**kwargs)


@dataclass
class JLSampler(Sampler):
    weights: torch.Tensor  # [V, D]
    k: int