ml_switcheroo.frameworks

Framework Adapters Package.

Automatically discovers and registers framework adapters by scanning this directory for modules. This allows for “Zero-Edit” extensibility: simply drop a new adapter file (e.g., tinygrad.py) into this folder, and it will be automatically registered.

This module exposes the registry helpers (get_adapter, available_frameworks) but relies on the side-effects of importing submodules to populate the internal _ADAPTER_REGISTRY.

Submodules

Classes

FrameworkAdapter

Protocol definition for a Framework Adapter.

Functions

register_framework(name)

get_adapter(→ Optional[FrameworkAdapter])

available_frameworks(→ List[str])

Package Contents

class ml_switcheroo.frameworks.FrameworkAdapter[source]

Bases: Protocol

Protocol definition for a Framework Adapter.

convert(data: Any) → Any[source]
property test_config: Dict[str, str]
property harness_imports: List[str]
get_harness_init_code() → str[source]
get_to_numpy_code() → str[source]
property search_modules: List[str]
property display_name: str
property ui_priority: int
property discovery_heuristics: Dict[str, List[str]]
property supported_tiers: List[ml_switcheroo.enums.SemanticTier]
property import_alias: Tuple[str, str]
property inherits_from: str | None
property structural_traits: ml_switcheroo.semantics.schema.StructuralTraits
property plugin_traits: ml_switcheroo.semantics.schema.PluginTraits
property rng_seed_methods: List[str]
property declared_magic_args: List[str]
property unsafe_submodules: Set[str]
get_device_syntax(device_type: str, device_index: str | None = None) → str[source]
get_device_check_syntax() → str[source]
get_rng_split_syntax(rng_var: str, key_var: str) → str[source]
get_serialization_syntax(op: str, file_arg: str, object_arg: str | None = None) → str[source]
get_serialization_imports() → List[str][source]
get_weight_conversion_imports() → List[str][source]
get_weight_load_code(path_var: str) → str[source]
get_tensor_to_numpy_expr(tensor_var: str) → str[source]
get_weight_save_code(state_var: str, path_var: str) → str[source]
get_doc_url(api_name: str) → str | None[source]
get_tiered_examples() → Dict[str, str][source]
property definitions: Dict[str, StandardMap]
property specifications: Dict[str, ml_switcheroo.semantics.schema.OperationDef]
property import_namespaces: Dict[str, Dict[str, str] | ImportConfig]
collect_api(category: StandardCategory) → List[ml_switcheroo.core.ghost.GhostRef][source]
apply_wiring(snapshot: Dict[str, Any]) → None[source]
ml_switcheroo.frameworks.register_framework(name: str)[source]
ml_switcheroo.frameworks.get_adapter(name: str) → FrameworkAdapter | None[source]
ml_switcheroo.frameworks.available_frameworks() → List[str][source]