src/tilelli/core/__init__.py

"""Tilelli core primitives — ternary quantizer + STE, ternary linear and
depthwise conv, diagonal SSM, sparse top-k attention, the 3-pathway
TilelliLiteLM (the deployed v4 chat model), and the parent multi-pathway
TilelliLM (used by the public training recipe — supports FP32 and ternary modes)."""

from tilelli.core.tilelli_lite import (
    PATHWAY_NAMES_LITE,
    TilelliLiteBlock,
    TilelliLiteLM,
)
from tilelli.core.tilelli_block import (
    PATHWAY_NAMES_3,
    PATHWAY_NAMES_5,
    TernaryFFN,
    TilelliBlock,
)
from tilelli.core.tilelli_lm import TilelliLM

__all__ = [
    "PATHWAY_NAMES_LITE",
    "PATHWAY_NAMES_3",
    "PATHWAY_NAMES_5",
    "TernaryFFN",
    "TilelliBlock",
    "TilelliBlock_Lite",
    "TilelliLiteLM",
    "TilelliLM",
]