cs249r_book/labs/mlsysbook_labs/__init__.py

"""Shared helpers for MLSysBook interactive trade-off labs."""

from .legacy_components import (
    Card,
    ComparisonRow,
    DecisionLog,
    FailureBanner,
    LatencyWaterfall,
    MathPeek,
    MetricRow,
    PredictionLock,
    RooflineVisualizer,
    StakeholderMessage,
)
from .catalog import LAB_CATALOG, get_lab_metadata
from .edge import (
    AdaptationStorageResult,
    EdgeDeviceProfile,
    EnergyDrainResult,
    FederatedCommunicationResult,
    TrainingMemoryBreakdown,
    adaptation_storage,
    edge_device_profile,
    energy_drain,
    federated_communication,
    training_memory_breakdown,
)
from .deployment import (
    ConstraintCheck,
    DeploymentEnvelopeResult,
    DeploymentSweepResult,
    DeploymentTrackProfile,
    MitigationResult,
    PlacementOption,
    deployment_mitigation,
    deployment_track_profile,
    evaluate_deployment_envelope,
    sweep_deployment_knob,
)
from .data_pipeline import (
    DataMovementStrategy,
    DataPipelineTrackProfile,
    MovementFrontierResult,
    PipelineArchitectureResult,
    PipelineResult,
    PipelineStageResult,
    data_pipeline_profile,
    evaluate_pipeline,
    movement_frontier,
    pipeline_architecture,
)
from .neural_compute import (
    MemoryCliffResult,
    NeuralComputeTrackProfile,
    OperationLedgerResult,
    OperatorDesignOption,
    OperatorDesignResult,
    memory_cliff,
    neural_compute_profile,
    operation_ledger,
    operator_design,
)
from .architecture import (
    ArchitectureCandidate,
    ArchitectureDecisionResult,
    ArchitectureEvaluation,
    ArchitectureScalingPoint,
    ArchitectureScalingResult,
    ArchitectureTrackProfile,
    architecture_decision,
    architecture_scaling_curve,
    architecture_signature,
    architecture_track_profile,
)
from .frameworks import (
    CompileBreakEvenResult,
    DispatchStackResult,
    FrameworkTrackProfile,
    RuntimeDecisionResult,
    RuntimeOption,
    compile_break_even,
    dispatch_stack,
    framework_track_profile,
    runtime_decision,
)
from .training import (
    TrainingFrontierPoint,
    TrainingFrontierResult,
    TrainingMemoryStackResult,
    TrainingPlanResult,
    TrainingStrategyOption,
    TrainingTrackProfile,
    training_frontier,
    training_memory_stack,
    training_plan,
    training_track_profile,
)
from .selection import (
    CoverageCellResult,
    CoverageProfileResult,
    CoverageSubgroup,
    DataPolicyDecisionResult,
    DataPolicyOption,
    DataSelectionTrackProfile,
    SelectionUtilityResult,
    coverage_profile,
    data_policy_decision,
    data_selection_profile,
    selection_frontier,
    selection_utility,
)
from .system_design import (
    SystemCurvePoint,
    SystemCurveResult,
    SystemDecisionResult,
    SystemDesignContext,
    SystemDesignControls,
    SystemDesignOption,
    SystemDesignProfile,
    SystemLedgerDecision,
    SystemOptionResult,
    render_system_design_lab,
    system_curve,
    system_decision,
    system_design_context,
    system_design_controls,
    system_design_ledger_summary,
    system_design_profile,
    system_frontier,
    system_option_result,
)
from .benchmarking import (
    AmdahlResult,
    BenchmarkTrackProfile,
    MetricGateResult,
    TailLatencyResult,
    ThermalSustainResult,
    amdahl_speedup,
    benchmark_track_profile,
    metric_gate,
    sustained_benchmark,
    tail_latency,
)
from .capstone import (
    ArchitectureMemoResult,
    CapstoneTrackProfile,
    LedgerDecision,
    LedgerReplayResult,
    SensitivityAuditResult,
    SensitivityAxis,
    architecture_memo,
    capstone_track_profile,
    replay_ledger,
    sensitivity_audit,
)
from .inference import (
    BatchingResult,
    CostCrossoverResult,
    InferenceEconomyProfile,
    ServingPlanResult,
    StateCapacityResult,
    batching_result,
    cost_crossover,
    inference_economy_profile,
    serving_plan,
    state_capacity,
)
from .migration import (
    baseline_big_takeaways,
    baseline_learning_objectives,
    build_migration_report,
    legacy_migration_panel,
    variant_source_trace,
)
from .ops import (
    DebtCascadeResult,
    DriftVisibilityResult,
    OpsPolicyResult,
    OpsTrackProfile,
    RetrainingCadenceResult,
    debt_cascade,
    drift_visibility,
    ops_policy,
    ops_track_profile,
    retraining_cadence,
)
from .reports import build_lab_report, report_export, report_export_panel, report_text_fallback
from .registry_refs import resolve_mlsysim_ref
from .responsibility import (
    CarbonBudgetResult,
    ExplanationOverheadResult,
    MetricConflictResult,
    ResponsibilityBudgetResult,
    ResponsibilityTrackProfile,
    carbon_budget,
    explanation_overhead,
    metric_conflict,
    responsibility_budget,
    responsibility_track_profile,
)
from .roofline import (
    FusionTrafficResult,
    GemmWorkload,
    RooflineHardwareProfile,
    RooflinePoint,
    fusion_traffic,
    gemm_workload,
    hardware_roofline_profile,
    roofline_point,
)
from .serving import (
    BatchingTaxResult,
    CacheCapacityResult,
    ColdStartResult,
    QueueingResult,
    ServingTrackProfile,
    batching_tax,
    cache_capacity,
    cold_start_latency,
    queueing_latency,
    serving_track_profile,
)
from .schemas import (
    DEFAULT_TRACKS,
    ChapterRecap,
    InstructorMetadata,
    LabTrackVariant,
    LabMetadata,
    LabReport,
    LedgerEntry,
    NuggetSpec,
    TrackProfile,
    TrackSpec,
)
from .state import DesignLedger, LedgerState
from .style import ACADEMIC_LAB_CSS, COLORS, LAB_CSS, apply_plotly_theme
from .tracks import (
    CANONICAL_TRACKS,
    DEFAULT_TRACK_ID,
    TRACK_ALIASES,
    get_track_profile,
    normalize_track_id,
    track_display_label,
    track_emoji,
    track_options,
    track_profile_map,
)
from .track_arcs import (
    LAB_ARC_STEPS,
    TRACK_ARCS,
    LabArcStep,
    TrackArc,
    get_lab_arc_step,
    get_track_arc,
    lab_arc_step_map,
    list_lab_arc_steps,
    list_track_arcs,
    track_arc_context_summary,
    track_arc_map,
    validate_track_arcs,
)
from .triad import (
    InterventionFrontierResult,
    TriadDiagnosisResult,
    TriadTrackProfile,
    diagnose_triad,
    intervention_frontier,
    triad_track_profile,
)
from .workflow import (
    ConstraintTaxResult,
    IterationFrontierResult,
    WorkflowGate,
    WorkflowPolicyResult,
    WorkflowTrackProfile,
    constraint_tax,
    iteration_frontier,
    workflow_policy,
    workflow_track_profile,
)
from .ui import (
    COMPLETION_STATES,
    action_box,
    advanced_knob_drawer,
    big_takeaways,
    chapter_recap,
    checkpoint_card,
    constraint_check,
    decision_flow,
    decision_card,
    evidence_summary,
    instructor_adoption_card,
    lab_map,
    lab_header,
    learning_objectives,
    micro_brief,
    nugget_shell,
    part_workflow,
    part_header,
    reflection_card,
    scenario_brief,
    scenario_thread,
    scenario_slice,
    source_trace,
    track_arc_context,
    track_context,
    track_selector,
    what_you_need_to_know,
)
from .variants import (
    ALL_LAB_IDS,
    LAB_TRACK_VARIANTS,
    PILOT_LAB_IDS,
    canonical_track_ids,
    get_lab_track_variant,
    list_lab_variants,
    variant_coverage,
)
from .versions import LEDGER_SCHEMA_VERSION, MLSYSBOOK_LABS_VERSION, REPORT_SCHEMA_VERSION, STABLE_RELEASE_CHANNEL

__all__ = [
    "ACADEMIC_LAB_CSS",
    "COLORS",
    "LAB_CSS",
    "apply_plotly_theme",
    "AdaptationStorageResult",
    "EdgeDeviceProfile",
    "EnergyDrainResult",
    "FederatedCommunicationResult",
    "TrainingMemoryBreakdown",
    "TrainingFrontierPoint",
    "TrainingFrontierResult",
    "TrainingMemoryStackResult",
    "TrainingPlanResult",
    "TrainingStrategyOption",
    "TrainingTrackProfile",
    "AmdahlResult",
    "BenchmarkTrackProfile",
    "MetricGateResult",
    "TailLatencyResult",
    "ThermalSustainResult",
    "ArchitectureMemoResult",
    "ArchitectureCandidate",
    "ArchitectureDecisionResult",
    "ArchitectureEvaluation",
    "ArchitectureScalingPoint",
    "ArchitectureScalingResult",
    "ArchitectureTrackProfile",
    "BatchingResult",
    "BatchingTaxResult",
    "CapstoneTrackProfile",
    "CacheCapacityResult",
    "ColdStartResult",
    "CompileBreakEvenResult",
    "ConstraintCheck",
    "ConstraintTaxResult",
    "CostCrossoverResult",
    "CoverageCellResult",
    "CoverageProfileResult",
    "CoverageSubgroup",
    "DebtCascadeResult",
    "DataMovementStrategy",
    "DataPolicyDecisionResult",
    "DataPolicyOption",
    "DataPipelineTrackProfile",
    "DataSelectionTrackProfile",
    "DeploymentEnvelopeResult",
    "DeploymentSweepResult",
    "DeploymentTrackProfile",
    "DriftVisibilityResult",
    "DispatchStackResult",
    "FrameworkTrackProfile",
    "InferenceEconomyProfile",
    "InterventionFrontierResult",
    "IterationFrontierResult",
    "MitigationResult",
    "OpsPolicyResult",
    "PlacementOption",
    "OpsTrackProfile",
    "QueueingResult",
    "CarbonBudgetResult",
    "ExplanationOverheadResult",
    "MetricConflictResult",
    "MemoryCliffResult",
    "MovementFrontierResult",
    "NeuralComputeTrackProfile",
    "OperationLedgerResult",
    "OperatorDesignOption",
    "OperatorDesignResult",
    "RetrainingCadenceResult",
    "RuntimeDecisionResult",
    "RuntimeOption",
    "PipelineArchitectureResult",
    "PipelineResult",
    "PipelineStageResult",
    "ResponsibilityBudgetResult",
    "ResponsibilityTrackProfile",
    "LedgerDecision",
    "LedgerReplayResult",
    "SensitivityAuditResult",
    "SensitivityAxis",
    "ServingTrackProfile",
    "ServingPlanResult",
    "SelectionUtilityResult",
    "StateCapacityResult",
    "SystemCurvePoint",
    "SystemCurveResult",
    "SystemDecisionResult",
    "SystemDesignContext",
    "SystemDesignControls",
    "SystemDesignOption",
    "SystemDesignProfile",
    "SystemLedgerDecision",
    "SystemOptionResult",
    "TriadDiagnosisResult",
    "TriadTrackProfile",
    "WorkflowGate",
    "WorkflowPolicyResult",
    "WorkflowTrackProfile",
    "adaptation_storage",
    "amdahl_speedup",
    "architecture_memo",
    "architecture_decision",
    "architecture_scaling_curve",
    "architecture_signature",
    "architecture_track_profile",
    "benchmark_track_profile",
    "batching_result",
    "batching_tax",
    "capstone_track_profile",
    "cache_capacity",
    "cold_start_latency",
    "cost_crossover",
    "carbon_budget",
    "coverage_profile",
    "compile_break_even",
    "constraint_tax",
    "data_pipeline_profile",
    "data_policy_decision",
    "data_selection_profile",
    "debt_cascade",
    "diagnose_triad",
    "deployment_mitigation",
    "deployment_track_profile",
    "drift_visibility",
    "dispatch_stack",
    "edge_device_profile",
    "energy_drain",
    "explanation_overhead",
    "evaluate_deployment_envelope",
    "evaluate_pipeline",
    "federated_communication",
    "framework_track_profile",
    "inference_economy_profile",
    "intervention_frontier",
    "iteration_frontier",
    "metric_gate",
    "metric_conflict",
    "memory_cliff",
    "movement_frontier",
    "neural_compute_profile",
    "operation_ledger",
    "operator_design",
    "ops_policy",
    "ops_track_profile",
    "pipeline_architecture",
    "queueing_latency",
    "replay_ledger",
    "retraining_cadence",
    "responsibility_budget",
    "responsibility_track_profile",
    "render_system_design_lab",
    "runtime_decision",
    "sensitivity_audit",
    "serving_plan",
    "serving_track_profile",
    "selection_frontier",
    "selection_utility",
    "state_capacity",
    "system_curve",
    "system_decision",
    "system_design_context",
    "system_design_controls",
    "system_design_ledger_summary",
    "system_design_profile",
    "system_frontier",
    "system_option_result",
    "sustained_benchmark",
    "sweep_deployment_knob",
    "tail_latency",
    "training_frontier",
    "training_memory_stack",
    "triad_track_profile",
    "training_memory_breakdown",
    "training_plan",
    "training_track_profile",
    "workflow_policy",
    "workflow_track_profile",
    "baseline_big_takeaways",
    "baseline_learning_objectives",
    "build_migration_report",
    "legacy_migration_panel",
    "variant_source_trace",
    "DesignLedger",
    "LedgerState",
    "Card",
    "ComparisonRow",
    "DecisionLog",
    "FailureBanner",
    "LatencyWaterfall",
    "MathPeek",
    "MetricRow",
    "PredictionLock",
    "RooflineVisualizer",
    "StakeholderMessage",
    "LabMetadata",
    "ChapterRecap",
    "TrackSpec",
    "TrackProfile",
    "LabTrackVariant",
    "NuggetSpec",
    "InstructorMetadata",
    "LedgerEntry",
    "LabReport",
    "LAB_CATALOG",
    "DEFAULT_TRACKS",
    "CANONICAL_TRACKS",
    "DEFAULT_TRACK_ID",
    "TRACK_ALIASES",
    "TRACK_ARCS",
    "LAB_ARC_STEPS",
    "MLSYSBOOK_LABS_VERSION",
    "REPORT_SCHEMA_VERSION",
    "LEDGER_SCHEMA_VERSION",
    "STABLE_RELEASE_CHANNEL",
    "normalize_track_id",
    "get_track_profile",
    "track_display_label",
    "track_emoji",
    "track_options",
    "track_profile_map",
    "TrackArc",
    "LabArcStep",
    "get_track_arc",
    "list_track_arcs",
    "track_arc_map",
    "get_lab_arc_step",
    "list_lab_arc_steps",
    "lab_arc_step_map",
    "track_arc_context_summary",
    "validate_track_arcs",
    "ALL_LAB_IDS",
    "LAB_TRACK_VARIANTS",
    "PILOT_LAB_IDS",
    "canonical_track_ids",
    "get_lab_track_variant",
    "list_lab_variants",
    "variant_coverage",
    "COMPLETION_STATES",
    "action_box",
    "lab_header",
    "learning_objectives",
    "chapter_recap",
    "scenario_brief",
    "scenario_thread",
    "decision_flow",
    "lab_map",
    "track_selector",
    "track_context",
    "track_arc_context",
    "part_header",
    "part_workflow",
    "what_you_need_to_know",
    "micro_brief",
    "scenario_slice",
    "nugget_shell",
    "constraint_check",
    "source_trace",
    "evidence_summary",
    "checkpoint_card",
    "big_takeaways",
    "advanced_knob_drawer",
    "reflection_card",
    "decision_card",
    "instructor_adoption_card",
    "build_lab_report",
    "report_export",
    "report_export_panel",
    "report_text_fallback",
    "get_lab_metadata",
    "resolve_mlsysim_ref",
    "FusionTrafficResult",
    "GemmWorkload",
    "RooflineHardwareProfile",
    "RooflinePoint",
    "fusion_traffic",
    "gemm_workload",
    "hardware_roofline_profile",
    "roofline_point",
]