"""Isolated, paper-faithful LoopQ reproduction components."""

from .quantization import (
    GroupwiseQuantizationResult,
    fake_quantize_groupwise,
    quantize_activation,
    quantize_weight,
)
from .las import HUGINN_LOOP_COUNT, OURO_LOOP_COUNT, LoopAwareActivationScales
from .transforms import SharedKroneckerTransform
from .sharing_gap import (
    HUGINN_PARAMETER_FRACTION,
    OURO_1_4B_GROUP_BUDGET,
    ProgressiveSelectionArtifact,
    SharingGapStatistics,
    progressive_select_by_group_count,
    progressive_select_huginn,
    sharing_gap_score,
)
from .cta import (
    DEFAULT_CTA_RANK,
    HUGINN_TRANSITION_COUNT,
    OURO_TRANSITION_COUNT,
    CrossLoopTransitionAdapter,
)
from .objective import (
    AdaptiveMuCache,
    TrajectoryLoss,
    adaptive_mu,
    topk_teacher_kl,
    trajectory_aware_loss,
)
from .calibrate import (
    CalibrationConfig,
    CalibrationParameterAllowlist,
    build_optimizer,
    build_parameter_allowlist,
    calibration_step,
    load_calibration_checkpoint,
    save_calibration_checkpoint,
)
from .export import (
    build_export_artifact,
    load_export_artifact,
    save_export_artifact,
    validate_export_artifact,
)

__all__ = [
    "GroupwiseQuantizationResult",
    "fake_quantize_groupwise",
    "quantize_activation",
    "quantize_weight",
    "HUGINN_LOOP_COUNT",
    "OURO_LOOP_COUNT",
    "LoopAwareActivationScales",
    "SharedKroneckerTransform",
    "HUGINN_PARAMETER_FRACTION",
    "OURO_1_4B_GROUP_BUDGET",
    "ProgressiveSelectionArtifact",
    "SharingGapStatistics",
    "progressive_select_by_group_count",
    "progressive_select_huginn",
    "sharing_gap_score",
    "DEFAULT_CTA_RANK",
    "HUGINN_TRANSITION_COUNT",
    "OURO_TRANSITION_COUNT",
    "CrossLoopTransitionAdapter",
    "AdaptiveMuCache",
    "TrajectoryLoss",
    "adaptive_mu",
    "topk_teacher_kl",
    "trajectory_aware_loss",
    "CalibrationConfig",
    "CalibrationParameterAllowlist",
    "build_optimizer",
    "build_parameter_allowlist",
    "calibration_step",
    "load_calibration_checkpoint",
    "save_calibration_checkpoint",
    "build_export_artifact",
    "load_export_artifact",
    "save_export_artifact",
    "validate_export_artifact",
]