Method prepare

src/fused_mm_sampling/core.py:679–690

(self)

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

    @classmethod
    def from_weights(
        cls,
        weights: torch.Tensor,  # [V, D]
        epsilon: float = 0.2,
    ) -> "JLSampler":
        k = optimal_k(n=weights.shape[0], epsilon=epsilon)
        print(f"JLSampler optimal k={k}")
        return cls(weights, k=k)

    def prepare(self) -> "JLSampler":