builtin

Built-in post-MIP node declarations.

Classes

AIPerfNode
DownstreamEvaluationNode
EvaluationNode
FilterNode
GlobalKDNode
ManualFilterNode
MaterializeNode
PTQNode

class AIPerfNode

Bases: PostMIPNode

capabilities: ClassVar[NodeCapabilities] = NodeCapabilities(kind=<NodeKind.EVALUATOR: 'evaluator'>, accepts=frozenset({<ArtifactKind.CHECKPOINT: 'checkpoint'>}), output=None, distributed=True, implemented=True, default_strategy='sharded')

classmethod render_report(node, payload)

type_name: ClassVar[str] = 'aiperf'

class DownstreamEvaluationNode

Bases: PostMIPNode

capabilities: ClassVar[NodeCapabilities] = NodeCapabilities(kind=<NodeKind.EVALUATOR: 'evaluator'>, accepts=frozenset({<ArtifactKind.CHECKPOINT: 'checkpoint'>}), output=None, distributed=True, implemented=True, default_strategy='sharded')

classmethod render_report(node, payload)

type_name: ClassVar[str] = 'downstream_evaluation'

class EvaluationNode

Bases: PostMIPNode

capabilities: ClassVar[NodeCapabilities] = NodeCapabilities(kind=<NodeKind.EVALUATOR: 'evaluator'>, accepts=frozenset({<ArtifactKind.CONFIG: 'config'>, <ArtifactKind.CHECKPOINT: 'checkpoint'>}), output=None, distributed=True, implemented=True, default_strategy='sharded')

classmethod render_report(node, payload)

type_name: ClassVar[str] = 'evaluation'

class FilterNode

Bases: PostMIPNode

capabilities: ClassVar[NodeCapabilities] = NodeCapabilities(kind=<NodeKind.SELECTOR: 'selector'>, accepts=frozenset({<ArtifactKind.CONFIG: 'config'>, <ArtifactKind.CHECKPOINT: 'checkpoint'>}), output=None, distributed=False, implemented=True, default_strategy='single')

classmethod metric_references(config)

Parameters:

config (Mapping[str, Any])

Return type:

tuple[str, …]

type_name: ClassVar[str] = 'filter'

classmethod validate_config(config)

Parameters:

config (Mapping[str, Any])

Return type:

None

class GlobalKDNode

Bases: PostMIPNode

capabilities: ClassVar[NodeCapabilities] = NodeCapabilities(kind=<NodeKind.TRANSFORMER: 'transformer'>, accepts=frozenset({<ArtifactKind.CHECKPOINT: 'checkpoint'>}), output=<ArtifactKind.CHECKPOINT: 'checkpoint'>, distributed=True, implemented=True, default_strategy='sharded')

classmethod render_report(node, payload)

type_name: ClassVar[str] = 'global_kd'

class ManualFilterNode

Bases: PostMIPNode

capabilities: ClassVar[NodeCapabilities] = NodeCapabilities(kind=<NodeKind.SELECTOR: 'selector'>, accepts=frozenset({<ArtifactKind.CONFIG: 'config'>, <ArtifactKind.CHECKPOINT: 'checkpoint'>}), output=None, distributed=False, implemented=True, default_strategy='single')

common_fields: ClassVar[frozenset[str]] = frozenset({'config', 'failure_policy', 'input', 'model_source', 'prompt', 'type'})

type_name: ClassVar[str] = 'manual_filter'

class MaterializeNode

Bases: PostMIPNode

capabilities: ClassVar[NodeCapabilities] = NodeCapabilities(kind=<NodeKind.TRANSFORMER: 'transformer'>, accepts=frozenset({<ArtifactKind.CONFIG: 'config'>, <ArtifactKind.CHECKPOINT: 'checkpoint'>}), output=<ArtifactKind.CHECKPOINT: 'checkpoint'>, distributed=True, implemented=True, default_strategy='sharded')

type_name: ClassVar[str] = 'materialize'

class PTQNode

Bases: PostMIPNode

capabilities: ClassVar[NodeCapabilities] = NodeCapabilities(kind=<NodeKind.TRANSFORMER: 'transformer'>, accepts=frozenset({<ArtifactKind.CHECKPOINT: 'checkpoint'>}), output=<ArtifactKind.CHECKPOINT: 'checkpoint'>, distributed=True, implemented=False, default_strategy='sharded')

type_name: ClassVar[str] = 'ptq'