inference/__init__.py

"""inference package — text generation and sampling."""

from inference.generate import generate, sample_logits
from inference.sampler  import Sampler

__all__ = [
    "generate",
    "sample_logits",
    "Sampler",
]