mixle.experimental.typed_runtime.structured_execution moduleΒΆ

Typed adapter for the stable exact model-parallel sufficient-statistic fold.

class StructuredEstimationReceipt(placement, observations, num_workers, worker_device_ids, execution_backend, parallel_node_ids, parallel_statistics_hash, reference_statistics_hash, parallel_model_hash, reference_model_hash, exact_parity, work, reference_seconds=None)[source]

Bases: object

Placement, work, sufficient-statistic, and model parity for one M-step.

Parameters:

  • placement (StructuredPlacementPlan)

  • observations (float)

  • num_workers (int)

  • worker_device_ids (tuple[str, ...])

  • execution_backend (str)

  • parallel_node_ids (tuple[str, ...])

  • parallel_statistics_hash (str)

  • reference_statistics_hash (str | None)

  • parallel_model_hash (str)

  • reference_model_hash (str | None)

  • exact_parity (bool | None)

  • work (WorkMeasurement)

  • reference_seconds (float | None)

as_dict()[source]

Return a JSON-compatible structured-execution receipt.

Return type:

dict[str, Any]

class StructuredEstimationResult(model, receipt)[source]

Bases: object

Estimated model and receipt; runtime model excluded from serialization.

Parameters:

  • model (ProbabilityDistribution)

  • receipt (StructuredEstimationReceipt)

run_structured_estimation_step(encoded_data, estimator, model, topology, *, weights=None, num_workers=None, verify_reference=True)[source]

Execute one exact typed model-axis E/M step and optionally verify serial parity.

Parameters:

  • encoded_data (Any)

  • estimator (ParameterEstimator)

  • model (ProbabilityDistribution)

  • topology (ClusterTopology)

  • weights (ndarray | None)

  • num_workers (int | None)

  • verify_reference (bool)

Return type:

StructuredEstimationResult