from __future__ import annotations

from .data import (
    LumiRingBuffer,
    MinHashDeduplicator,
    PIIMasker,
    QualityFilter,
    ResumablePipeline,
    StateTracker,
    StreamCoordinate,
    StreamDeduplicator,
    StringTensor,
    TextCleaner,
    TokenRingBuffer,
    clean_batch,
    clean_text,
    create_resumable_collate,
    data_cpu_isa,
    filter_quality,
    fused_process_batch,
    FusedPipeline,
    hash_record,
    is_high_quality,
    jaccard_similarity,
    mask_pii,
    postprocess_and_dedup_records,
    serialize_to_arrow,
    serialize_to_jsonl,
    serialize_to_parquet,
    suggest_closest_matches,
    get_version,
    list_versions,
    set_version,
)
from .optimizer import LuminaV
from . import nn
from .nn import (
    ChunkedLinearCrossEntropy,
    FusedLinearCrossEntropy,
    FusedRaEMA,
    HybridRationalMoE,
    PosRaGLU,
    RaAttention,
    RaAttn,
    RaBatchNorm2d,
    RaEMA,
    RaGELU,
    RaGLU,
    RaGQA,
    RaLinearCrossEntropy,
    RaMoE,
    RaNorm,
    RaTanh,
    RationalEMA,
    RationalGQA,
    RationalMoE,
    TriGLU,
    chunked_linear_cross_entropy,
    flash_ra_attention,
    fused_linear_cross_entropy,
    fused_ra_ema,
    permute_tokens_by_expert,
    pos_ra_gate,
    pos_ra_glu,
    ra_attention,
    ra_ema,
    ra_gelu,
    ra_linear_cross_entropy,
    ra_softmax,
    ra_tanh,
)

__all__ = [
    "LumiRingBuffer",
    "LuminaV",
    "MinHashDeduplicator",
    "PIIMasker",
    "QualityFilter",
    "ResumablePipeline",
    "StateTracker",
    "StreamCoordinate",
    "StreamDeduplicator",
    "StringTensor",
    "TextCleaner",
    "TokenRingBuffer",
    "__version__",
    "clean_batch",
    "clean_text",
    "create_resumable_collate",
    "data_cpu_isa",
    "filter_quality",
    "fused_process_batch",
    "FusedPipeline",
    "hash_record",
    "is_high_quality",
    "jaccard_similarity",
    "mask_pii",
    "postprocess_and_dedup_records",
    "serialize_to_arrow",
    "serialize_to_jsonl",
    "serialize_to_parquet",
    "suggest_closest_matches",
    "get_version",
    "list_versions",
    "set_version",
    # Neural Network Components (Ra-Family)
    "nn",
    "RaNorm",
    "RaBatchNorm2d",
    "RaTanh",
    "ra_tanh",
    "RaGELU",
    "ra_gelu",
    "pos_ra_gate",
    "pos_ra_glu",
    "PosRaGLU",
    "RaGLU",
    "TriGLU",
    "RaAttention",
    "RaAttn",
    "RaGQA",
    "RationalGQA",
    "ra_attention",
    "flash_ra_attention",
    "ra_softmax",
    "RaEMA",
    "RationalEMA",
    "FusedRaEMA",
    "fused_ra_ema",
    "ra_ema",
    "FusedLinearCrossEntropy",
    "ChunkedLinearCrossEntropy",
    "RaLinearCrossEntropy",
    "fused_linear_cross_entropy",
    "chunked_linear_cross_entropy",
    "ra_linear_cross_entropy",
    "RationalMoE",
    "HybridRationalMoE",
    "RaMoE",
    "permute_tokens_by_expert",
]
from ._version import __version__