Index

_

__class_vars__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __private_attributes__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_complete__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_computed_fields__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_core_schema__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_custom_init__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_decorators__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_extra__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_fields__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_fields_set__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_generic_metadata__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_parent_namespace__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_post_init__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_private__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute)

__pydantic_root_model__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_serializer__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __pydantic_validator__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __signature__ (ml_switcheroo.semantics.schema.FrameworkAlias attribute) __version__ (in module ml_switcheroo) _current_violations (ml_switcheroo.analysis.purity.PurityScanner attribute) _dynamic_impurity_methods (ml_switcheroo.analysis.purity.PurityScanner attribute) _global_rng_methods (ml_switcheroo.analysis.purity.PurityScanner attribute) _in_annotation (ml_switcheroo.core.rewriter.structure_types.TypeStructureMixin attribute) _IO_FUNCTIONS (ml_switcheroo.analysis.purity.PurityScanner attribute) _local_aliases (ml_switcheroo.testing.linter.StructuralLinter attribute) _MUTATION_METHODS (ml_switcheroo.analysis.purity.PurityScanner attribute) _package_cache (ml_switcheroo.discovery.inspector.ApiInspector attribute) _processed_fws (ml_switcheroo.generated_tests.generator.TestGenerator attribute) _scope_stack (ml_switcheroo.analysis.lifecycle.InitializationTracker attribute)

A

ACTIVATION (ml_switcheroo.frameworks.base.StandardCategory attribute) adapt_to_framework() (ml_switcheroo.testing.fuzzer.InputFuzzer method) adapter (ml_switcheroo.core.discovery.SimulatedReflection attribute) add_variant() (ml_switcheroo.discovery.consensus.CandidateStandard method) alias (ml_switcheroo.core.dsl.ImportReq attribute) (ml_switcheroo.semantics.schema.FrameworkTraits attribute) alias_map (ml_switcheroo.core.import_fixer.ImportFixer attribute) aliases (ml_switcheroo.discovery.harvester.ImportScanner attribute) (ml_switcheroo.discovery.harvester.TargetCallVisitor attribute) align_signatures() (ml_switcheroo.cli.commands.ConsensusEngine method) (ml_switcheroo.discovery.consensus.ConsensusEngine method) allowed_roots (ml_switcheroo.analysis.audit.CoverageScanner attribute) ANALYSIS_WARNING (ml_switcheroo.core.tracer.TraceEventType attribute) annotation (ml_switcheroo.core.ghost.GhostParam attribute) api (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) (ml_switcheroo.semantics.schema.Variant attribute) api_path (ml_switcheroo.core.ghost.GhostRef attribute) ApiInspector (class in ml_switcheroo.discovery.inspector) apply_index_select() (in module ml_switcheroo.core.rewriter.calls.transformers) apply_output_adapter() (in module ml_switcheroo.core.rewriter.calls.transformers) apply_wiring() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method)

ARG_ALIASES (ml_switcheroo.cli.commands.ConsensusEngine attribute) (ml_switcheroo.discovery.consensus.ConsensusEngine attribute) arg_mappings (ml_switcheroo.discovery.consensus.CandidateStandard attribute) arg_values (ml_switcheroo.core.dsl.FrameworkVariant attribute) ArgInjectorType (in module ml_switcheroo.core.hooks) args (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) (ml_switcheroo.semantics.schema.Variant attribute) arity_penalty (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) ARRAY_API (ml_switcheroo.enums.SemanticTier attribute), [1] ArrayApiSpecImporter (class in ml_switcheroo.importers.array_api_reader) AST_MUTATION (ml_switcheroo.core.tracer.TraceEventType attribute) ASTEngine (class in ml_switcheroo) (class in ml_switcheroo.core.engine) AttributeMixin (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.attributes) AutoWireSpec (class in ml_switcheroo.core.hooks) available_frameworks() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.frameworks)

B

BaseRewriter (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.base) BatchValidator (class in ml_switcheroo.cli.commands) (class in ml_switcheroo.testing.batch_runner)

BLOCK (ml_switcheroo.core.dsl.PluginType attribute) body_node (injector_plugin.BodyExtractor attribute) BodyExtractor (class in injector_plugin)

C

CALL (ml_switcheroo.core.dsl.PluginType attribute) CallMixin (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.calls) (class in ml_switcheroo.core.rewriter.calls.mixer) CandidateStandard (class in ml_switcheroo.discovery.consensus) capture_eval_state() (in module ml_switcheroo.plugins.state_flag_injection) capture_node_source() (in module ml_switcheroo.utils.node_diff) casts (ml_switcheroo.core.dsl.FrameworkVariant attribute) check() (ml_switcheroo.testing.linter.StructuralLinter method) ClassStructureMixin (class in ml_switcheroo.core.rewriter.structure_class) clear_hooks() (in module ml_switcheroo.core.hooks) cluster() (ml_switcheroo.cli.commands.ConsensusEngine method) (ml_switcheroo.discovery.consensus.ConsensusEngine method) code (ml_switcheroo.ConversionResult attribute) (ml_switcheroo.core.engine.ConversionResult attribute) CodeExtractor (class in ml_switcheroo.utils.code_extractor) collect_api() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) CompatibilityMatrix (class in ml_switcheroo.cli.matrix) complexity (ml_switcheroo.core.dsl.OperationDef attribute) compute_permutation() (in module ml_switcheroo.core.rewriter.calls.utils) config (ml_switcheroo.core.rewriter.base.BaseRewriter attribute) (ml_switcheroo.core.rewriter.BaseRewriter attribute) config() (ml_switcheroo.core.hooks.HookContext method) ConsensusEngine (class in ml_switcheroo.cli.commands) (class in ml_switcheroo.discovery.consensus)

console (in module ml_switcheroo.utils.console) (ml_switcheroo.cli.commands.FrameworkSyncer attribute), [1] (ml_switcheroo.cli.matrix.CompatibilityMatrix attribute) (ml_switcheroo.cli.wizard.MappingWizard attribute) (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) (ml_switcheroo.discovery.syncer.FrameworkSyncer attribute), [1] CONTEXT (ml_switcheroo.core.dsl.OpType attribute) ContextMixin (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.contexts) ControlFlowMixin (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.control_flow) ConversionResult (class in ml_switcheroo) (class in ml_switcheroo.core.engine) convert() (in module ml_switcheroo) (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) convert_load_state_dict() (in module ml_switcheroo.plugins.state_container) convert_parameters() (in module ml_switcheroo.plugins.state_container) convert_register_buffer() (in module ml_switcheroo.plugins.state_container) convert_register_parameter() (in module ml_switcheroo.plugins.state_container) convert_state_dict() (in module ml_switcheroo.plugins.state_container) CoverageScanner (class in ml_switcheroo.analysis.audit) ctx (ml_switcheroo.core.rewriter.base.BaseRewriter attribute) (ml_switcheroo.core.rewriter.BaseRewriter attribute) current_op_id (ml_switcheroo.core.hooks.HookContext attribute)

D

data (ml_switcheroo.cli.commands.SemanticsManager attribute) (ml_switcheroo.semantics.manager.SemanticsManager attribute) (ml_switcheroo.SemanticsManager attribute) DECORATOR (ml_switcheroo.core.dsl.OpType attribute) DecoratorMixin (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.decorators) default (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.ghost.GhostParam attribute) definitions (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) DependencyScanner (class in ml_switcheroo.analysis.dependencies) deprecated (ml_switcheroo.core.dsl.OperationDef attribute) description (ml_switcheroo.core.dsl.OperationDef attribute) (ml_switcheroo.core.tracer.TraceEvent attribute) (ml_switcheroo.semantics.schema.OpDefinition attribute) device_constraints (ml_switcheroo.core.dsl.OperationDef attribute) diff_nodes() (in module ml_switcheroo.utils.node_diff) differentiable (ml_switcheroo.core.dsl.OperationDef attribute) discover() (ml_switcheroo.core.discovery.SimulatedReflection method)

DISCOVERED_FILENAMES (in module ml_switcheroo.semantics.manager) discovery_heuristics (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) dispatch_rules (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.semantics.schema.Variant attribute) display_name (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter attribute) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter attribute) (ml_switcheroo.frameworks.keras.KerasAdapter attribute) (ml_switcheroo.frameworks.mlx.MLXAdapter attribute) (ml_switcheroo.frameworks.numpy.NumpyAdapter attribute) (ml_switcheroo.frameworks.paxml.PaxmlAdapter attribute) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter attribute) (ml_switcheroo.frameworks.torch.TorchAdapter attribute) doc (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.dsl.PluginScaffoldDef attribute) docstring (ml_switcheroo.core.ghost.GhostRef attribute) documentation_url (ml_switcheroo.core.dsl.OperationDef attribute) dtype (ml_switcheroo.core.dsl.ParameterDef attribute)

E

effective_source (ml_switcheroo.config.RuntimeConfig property) (ml_switcheroo.RuntimeConfig property) effective_target (ml_switcheroo.config.RuntimeConfig property) (ml_switcheroo.RuntimeConfig property) END_MARKER (ml_switcheroo.core.escape_hatch.EscapeHatch attribute) end_phase() (ml_switcheroo.core.tracer.TraceLogger method) EQ (ml_switcheroo.enums.LogicOp attribute) EquivalenceRunner (class in ml_switcheroo.testing.runner) errors (ml_switcheroo.ConversionResult attribute) (ml_switcheroo.core.engine.ConversionResult attribute)

EscapeHatch (class in ml_switcheroo.core.escape_hatch) evaluate_dispatch_rules() (in module ml_switcheroo.core.rewriter.calls.dispatch) existing_args (ml_switcheroo.core.rewriter.types.SignatureContext attribute) export() (ml_switcheroo.core.tracer.TraceLogger method) extends (ml_switcheroo.semantics.schema.FrameworkTraits attribute) extract_class() (ml_switcheroo.utils.code_extractor.CodeExtractor static method) extractor (ml_switcheroo.testing.harness_generator.HarnessGenerator attribute), [1] EXTRAS (ml_switcheroo.enums.SemanticTier attribute), [1]

F

files (in module ml_switcheroo.semantics.paths) filter_common() (ml_switcheroo.cli.commands.ConsensusEngine method) (ml_switcheroo.discovery.consensus.ConsensusEngine method) flax_nnx (in module ml_switcheroo.frameworks.flax_nnx) FlaxNNXAdapter (class in ml_switcheroo.frameworks.flax_nnx) (in module ml_switcheroo.frameworks) forbidden_roots (ml_switcheroo.testing.linter.StructuralLinter attribute), [1] forward_method (ml_switcheroo.semantics.schema.StructuralTraits attribute) found (injector_fw.FrameworkInjector attribute), [1] (injector_plugin.BodyExtractor attribute) (injector_spec.StandardsInjector attribute), [1] (ml_switcheroo.core.scanners.SimpleNameScanner attribute) found_usages (ml_switcheroo.core.scanners.UsageScanner attribute) framework (ml_switcheroo.core.discovery.SimulatedReflection attribute) framework_configs (ml_switcheroo.cli.commands.SemanticsManager attribute) (ml_switcheroo.semantics.manager.SemanticsManager attribute) (ml_switcheroo.SemanticsManager attribute)

FrameworkAdapter (class in ml_switcheroo.frameworks) (class in ml_switcheroo.frameworks.base) FrameworkAlias (class in ml_switcheroo.semantics.schema) FrameworkInjector (class in injector_fw) frameworks (ml_switcheroo.semantics.schema.SemanticsFile attribute) FrameworkSyncer (class in ml_switcheroo.cli.commands) (class in ml_switcheroo.discovery.syncer) FrameworkTraits (class in ml_switcheroo.semantics.schema) FrameworkVariant (class in ml_switcheroo.core.dsl) from_file (ml_switcheroo.semantics.schema.OpDefinition attribute) func_name (injector_plugin.BodyExtractor attribute) FuncStructureMixin (class in ml_switcheroo.core.rewriter.structure_func) FUNCTION (ml_switcheroo.core.dsl.OpType attribute) fuzzer (ml_switcheroo.testing.runner.EquivalenceRunner attribute), [1]

G

generate() (injector_plugin.PluginGenerator method) (ml_switcheroo.generated_tests.generator.TestGenerator method) (ml_switcheroo.testing.harness_generator.HarnessGenerator method) (ml_switcheroo.utils.doc_gen.MigrationGuideGenerator method) generate_inputs() (ml_switcheroo.testing.fuzzer.InputFuzzer method) get_adapter() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.frameworks) (in module ml_switcheroo.frameworks.base) get_all_hook_metadata() (in module ml_switcheroo.core.hooks) get_all_rng_methods() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) get_console() (in module ml_switcheroo.utils.console) get_dataloader_semantics() (in module ml_switcheroo.frameworks.common.data) get_definition() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) get_definition_by_id() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) get_device_syntax() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_example_code() (ml_switcheroo.frameworks.base.FrameworkAdapter class method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter class method) (ml_switcheroo.frameworks.FrameworkAdapter class method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter class method) (ml_switcheroo.frameworks.keras.KerasAdapter class method) (ml_switcheroo.frameworks.mlx.MLXAdapter class method) (ml_switcheroo.frameworks.numpy.NumpyAdapter class method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter class method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter class method) (ml_switcheroo.frameworks.torch.TorchAdapter class method) get_framework_aliases() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) get_framework_config() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) get_framework_priority_order() (in module ml_switcheroo.config)

get_full_name() (in module ml_switcheroo.core.scanners) get_hook() (in module ml_switcheroo.core.hooks) get_import_map() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) get_json() (ml_switcheroo.cli.matrix.CompatibilityMatrix method) get_known_apis() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) get_result() (ml_switcheroo.core.scanners.UsageScanner method) get_serialization_imports() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_serialization_syntax() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_shim_code() (in module ml_switcheroo.frameworks.common.data) get_test_template() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) get_tiered_examples() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_tracer() (in module ml_switcheroo.core.tracer) GHOST (ml_switcheroo.frameworks.base.InitMode attribute) GhostInspector (class in ml_switcheroo.core.ghost) GhostParam (class in ml_switcheroo.core.ghost) GhostRef (class in ml_switcheroo.core.ghost) GT (ml_switcheroo.enums.LogicOp attribute) GTE (ml_switcheroo.enums.LogicOp attribute)

H

handle_audit() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers.audit) handle_ci() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.verify) handle_convert() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers.convert) handle_define() (in module ml_switcheroo.cli.handlers.define) handle_docs() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.dev) handle_gen_tests() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.dev) handle_harvest() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.learning) handle_import_spec() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.discovery) handle_matrix() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.dev) handle_scaffold() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.discovery)

handle_snapshot() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.snapshots) handle_sync() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.snapshots) handle_sync_standards() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.discovery) handle_wizard() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.learning) HARNESS_TEMPLATE (in module ml_switcheroo.testing.harness_generator_template) HarnessGenerator (class in ml_switcheroo.testing.harness_generator) harvest_file() (ml_switcheroo.discovery.harvester.SemanticHarvester method) has_arg() (ml_switcheroo.core.ghost.GhostRef method) has_errors (ml_switcheroo.ConversionResult property) (ml_switcheroo.core.engine.ConversionResult property) has_varargs (ml_switcheroo.core.ghost.GhostRef attribute) HELPER_LOGIC (in module injector_plugin) HookContext (class in ml_switcheroo.core.hooks) HookFunction (in module ml_switcheroo.core.hooks) hydrate() (ml_switcheroo.core.ghost.GhostInspector static method)

I

id (ml_switcheroo.core.tracer.TraceEvent attribute) if_arg (ml_switcheroo.core.dsl.Rule attribute) (ml_switcheroo.semantics.schema.Rule attribute) IGNORED_SUFFIXES (ml_switcheroo.cli.commands.ConsensusEngine attribute) (ml_switcheroo.discovery.consensus.ConsensusEngine attribute) IMPORT_ACTION (ml_switcheroo.core.tracer.TraceEventType attribute) import_alias (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) import_data (ml_switcheroo.cli.commands.SemanticsManager attribute) (ml_switcheroo.semantics.manager.SemanticsManager attribute) (ml_switcheroo.SemanticsManager attribute) import_namespaces (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) ImportFixer (class in ml_switcheroo.core.import_fixer) ImportReq (class in ml_switcheroo.core.dsl) imports (ml_switcheroo.semantics.schema.SemanticsFile attribute) ImportScanner (class in ml_switcheroo.discovery.harvester) impurity_methods (ml_switcheroo.semantics.schema.StructuralTraits attribute) IN (ml_switcheroo.enums.LogicOp attribute) infer_tier_from_priority() (in module ml_switcheroo.semantics.merging) inherits_from (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter attribute) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter attribute) (ml_switcheroo.frameworks.keras.KerasAdapter attribute) (ml_switcheroo.frameworks.mlx.MLXAdapter attribute) (ml_switcheroo.frameworks.numpy.NumpyAdapter attribute) (ml_switcheroo.frameworks.paxml.PaxmlAdapter attribute) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter attribute) (ml_switcheroo.frameworks.torch.TorchAdapter attribute)

init_method_name (ml_switcheroo.semantics.schema.StructuralTraits attribute) INIT_NAME_MAP (in module ml_switcheroo.core.rewriter.initializers) InitializationTracker (class in ml_switcheroo.analysis.lifecycle) InitializerMixin (class in ml_switcheroo.core.rewriter.initializers) InitMode (class in ml_switcheroo.frameworks.base) inject_args (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) inject_custom_api_call() (in module ml_switcheroo.core.rewriter.calls.utils) inject_magic_args (ml_switcheroo.semantics.schema.StructuralTraits attribute) inject_permute_call() (in module ml_switcheroo.core.rewriter.calls.utils) inject_preamble() (ml_switcheroo.core.hooks.HookContext method) inject_prng_threading() (in module ml_switcheroo.plugins.rng_threading) inject_rngs_kwarg() (in module ml_switcheroo.core.rewriter.calls.utils) inject_signature_arg() (ml_switcheroo.core.hooks.HookContext method) inject_training_flag_call() (in module ml_switcheroo.plugins.state_flag_injection) injected_args (ml_switcheroo.core.rewriter.types.SignatureContext attribute) injector_fw module injector_plugin module injector_spec module InputFuzzer (class in ml_switcheroo.testing.fuzzer) inspect() (ml_switcheroo.core.ghost.GhostInspector static method) (ml_switcheroo.discovery.inspector.ApiInspector method) INSPECTION (ml_switcheroo.core.tracer.TraceEventType attribute) inspector (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) INTERNAL_OPS (in module ml_switcheroo.semantics.standards_internal) is_builtin() (in module ml_switcheroo.core.rewriter.calls.utils) is_functional_apply() (in module ml_switcheroo.core.rewriter.calls.utils) is_init (ml_switcheroo.core.rewriter.types.SignatureContext attribute) is_inplace (ml_switcheroo.core.dsl.OperationDef attribute) is_module_method (ml_switcheroo.core.rewriter.types.SignatureContext attribute) is_super_call() (in module ml_switcheroo.core.rewriter.calls.utils) is_val (ml_switcheroo.core.dsl.Rule attribute) (ml_switcheroo.semantics.schema.Rule attribute) is_variadic (ml_switcheroo.core.dsl.ParameterDef attribute) is_verified() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method)

J

jax (in module ml_switcheroo.frameworks.jax) jax_test_config (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin property) JaxAdapter (in module ml_switcheroo.frameworks) (in module ml_switcheroo.frameworks.jax)

JaxCoreAdapter (class in ml_switcheroo.frameworks.jax) JAXStackMixin (class in ml_switcheroo.frameworks.common.jax_stack) jit_static_args (ml_switcheroo.semantics.schema.StructuralTraits attribute)

K

keras (in module ml_switcheroo.frameworks.keras) KerasAdapter (class in ml_switcheroo.frameworks.keras) KEY_MAPPER_SOURCE (in module ml_switcheroo.plugins.checkpoint_keys)

kind (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.ghost.GhostParam attribute) (ml_switcheroo.core.ghost.GhostRef attribute) kwargs_map (ml_switcheroo.core.dsl.FrameworkVariant attribute)

L

LAYER (ml_switcheroo.frameworks.base.StandardCategory attribute) layout_map (ml_switcheroo.core.dsl.FrameworkVariant attribute) leave_Annotation() (ml_switcheroo.core.rewriter.structure_types.TypeStructureMixin method) leave_Assign() (injector_spec.StandardsInjector method) (ml_switcheroo.core.rewriter.AttributeMixin method) (ml_switcheroo.core.rewriter.attributes.AttributeMixin method) (ml_switcheroo.core.rewriter.CallMixin method) (ml_switcheroo.core.rewriter.calls.CallMixin method) (ml_switcheroo.core.rewriter.calls.mixer.CallMixin method) leave_Attribute() (ml_switcheroo.core.import_fixer.ImportFixer method) (ml_switcheroo.core.rewriter.AttributeMixin method) (ml_switcheroo.core.rewriter.attributes.AttributeMixin method) leave_Call() (ml_switcheroo.core.rewriter.CallMixin method) (ml_switcheroo.core.rewriter.calls.CallMixin method) (ml_switcheroo.core.rewriter.calls.mixer.CallMixin method) (ml_switcheroo.core.rewriter.initializers.InitializerMixin method) leave_ClassDef() (injector_fw.FrameworkInjector method) (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.core.rewriter.structure_class.ClassStructureMixin method) leave_Decorator() (ml_switcheroo.core.rewriter.DecoratorMixin method) (ml_switcheroo.core.rewriter.decorators.DecoratorMixin method) leave_For() (ml_switcheroo.core.rewriter.control_flow.ControlFlowMixin method) (ml_switcheroo.core.rewriter.ControlFlowMixin method) leave_FunctionDef() (injector_fw.FrameworkInjector method) (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.core.rewriter.structure_func.FuncStructureMixin method) leave_Import() (ml_switcheroo.core.import_fixer.ImportFixer method) (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) leave_ImportFrom() (ml_switcheroo.core.import_fixer.ImportFixer method) (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) leave_Module() (injector_fw.FrameworkInjector method) (ml_switcheroo.core.import_fixer.ImportFixer method) (ml_switcheroo.core.rewriter.base.BaseRewriter method) (ml_switcheroo.core.rewriter.BaseRewriter method)

leave_Name() (ml_switcheroo.core.rewriter.structure_types.TypeStructureMixin method) (ml_switcheroo.plugins.static_unroll.LoopVarReplacer method) leave_Return() (injector_fw.FrameworkInjector method) leave_SimpleStatementLine() (ml_switcheroo.analysis.purity.PurityScanner method) (ml_switcheroo.core.rewriter.base.BaseRewriter method) (ml_switcheroo.core.rewriter.BaseRewriter method) leave_With() (ml_switcheroo.core.rewriter.ContextMixin method) (ml_switcheroo.core.rewriter.contexts.ContextMixin method) lifecycle_strip_methods (ml_switcheroo.semantics.schema.StructuralTraits attribute) lifecycle_warn_methods (ml_switcheroo.semantics.schema.StructuralTraits attribute) LIVE (ml_switcheroo.frameworks.base.InitMode attribute) load() (ml_switcheroo.config.RuntimeConfig class method) (ml_switcheroo.RuntimeConfig class method) load_plugins() (in module ml_switcheroo.core.hooks) load_snapshot_for_adapter() (in module ml_switcheroo.frameworks.base) load_validation_report() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) log_diff() (in module ml_switcheroo.core.rewriter.calls.utils) log_error() (in module ml_switcheroo.utils.console) log_info() (in module ml_switcheroo.utils.console) log_inspection() (ml_switcheroo.core.tracer.TraceLogger method) log_match() (ml_switcheroo.core.tracer.TraceLogger method) log_mutation() (ml_switcheroo.core.tracer.TraceLogger method) log_success() (in module ml_switcheroo.utils.console) log_warning() (in module ml_switcheroo.utils.console) (ml_switcheroo.core.tracer.TraceLogger method) logger (in module ml_switcheroo.core.discovery) LogicOp (class in ml_switcheroo.enums) lookup_api() (ml_switcheroo.core.hooks.HookContext method) lookup_signature() (ml_switcheroo.core.hooks.HookContext method) LoopVarReplacer (class in ml_switcheroo.plugins.static_unroll) LOSS (ml_switcheroo.frameworks.base.StandardCategory attribute) LT (ml_switcheroo.enums.LogicOp attribute) LTE (ml_switcheroo.enums.LogicOp attribute)

M

macro_template (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) main() (in module ml_switcheroo.cli.__main__) mappings (ml_switcheroo.discovery.harvester.TargetCallVisitor attribute) MappingWizard (class in ml_switcheroo.cli.wizard) mark_failure() (ml_switcheroo.core.escape_hatch.EscapeHatch static method) MATCH_SEMANTICS (ml_switcheroo.core.tracer.TraceEventType attribute) max (ml_switcheroo.core.dsl.ParameterDef attribute) max_depth (ml_switcheroo.testing.fuzzer.InputFuzzer attribute) MAX_RECURSION_DEPTH (ml_switcheroo.testing.fuzzer.InputFuzzer attribute) max_version (ml_switcheroo.core.dsl.FrameworkVariant attribute) merge_frameworks() (in module ml_switcheroo.semantics.merging) merge_imports() (in module ml_switcheroo.semantics.merging) merge_overlay_data() (in module ml_switcheroo.semantics.merging) merge_tier_data() (in module ml_switcheroo.semantics.merging) metadata (ml_switcheroo.core.hooks.HookContext attribute) (ml_switcheroo.core.tracer.TraceEvent attribute) MigrationGuideGenerator (class in ml_switcheroo.utils.doc_gen) min (ml_switcheroo.core.dsl.ParameterDef attribute) min_version (ml_switcheroo.core.dsl.FrameworkVariant attribute) missing_message (ml_switcheroo.core.dsl.FrameworkVariant attribute) ml_switcheroo module ml_switcheroo.__main__ module ml_switcheroo.analysis module ml_switcheroo.analysis.audit module ml_switcheroo.analysis.dependencies module ml_switcheroo.analysis.lifecycle module ml_switcheroo.analysis.purity module ml_switcheroo.cli module ml_switcheroo.cli.__main__ module ml_switcheroo.cli.commands module ml_switcheroo.cli.handlers module ml_switcheroo.cli.handlers.audit module ml_switcheroo.cli.handlers.convert module ml_switcheroo.cli.handlers.define module ml_switcheroo.cli.handlers.dev module ml_switcheroo.cli.handlers.discovery module ml_switcheroo.cli.handlers.learning module ml_switcheroo.cli.handlers.snapshots module ml_switcheroo.cli.handlers.verify module ml_switcheroo.cli.matrix module ml_switcheroo.cli.wizard module ml_switcheroo.config module ml_switcheroo.core module ml_switcheroo.core.discovery module ml_switcheroo.core.dsl module ml_switcheroo.core.engine module ml_switcheroo.core.escape_hatch module ml_switcheroo.core.ghost module ml_switcheroo.core.hooks module ml_switcheroo.core.import_fixer module ml_switcheroo.core.rewriter module ml_switcheroo.core.rewriter.attributes module ml_switcheroo.core.rewriter.base module ml_switcheroo.core.rewriter.calls module ml_switcheroo.core.rewriter.calls.dispatch module ml_switcheroo.core.rewriter.calls.mixer module ml_switcheroo.core.rewriter.calls.transformers module ml_switcheroo.core.rewriter.calls.utils module ml_switcheroo.core.rewriter.contexts module ml_switcheroo.core.rewriter.control_flow module ml_switcheroo.core.rewriter.decorators module ml_switcheroo.core.rewriter.initializers module ml_switcheroo.core.rewriter.normalization module ml_switcheroo.core.rewriter.structure module ml_switcheroo.core.rewriter.structure_class module ml_switcheroo.core.rewriter.structure_func module ml_switcheroo.core.rewriter.structure_types module ml_switcheroo.core.rewriter.types module ml_switcheroo.core.scanners module ml_switcheroo.core.tracer module ml_switcheroo.discovery module ml_switcheroo.discovery.consensus module ml_switcheroo.discovery.harvester module ml_switcheroo.discovery.inspector module ml_switcheroo.discovery.scaffolder module ml_switcheroo.discovery.syncer module ml_switcheroo.enums module ml_switcheroo.frameworks module ml_switcheroo.frameworks.base module ml_switcheroo.frameworks.common module ml_switcheroo.frameworks.common.data module ml_switcheroo.frameworks.common.jax_stack module ml_switcheroo.frameworks.flax_nnx module ml_switcheroo.frameworks.jax module ml_switcheroo.frameworks.keras module ml_switcheroo.frameworks.mlx module ml_switcheroo.frameworks.numpy module ml_switcheroo.frameworks.optax_shim module ml_switcheroo.frameworks.paxml module ml_switcheroo.frameworks.tensorflow module ml_switcheroo.frameworks.torch module ml_switcheroo.generated_tests module ml_switcheroo.generated_tests.generator module ml_switcheroo.importers module ml_switcheroo.importers.array_api_reader module ml_switcheroo.importers.onnx_reader module ml_switcheroo.plugins module ml_switcheroo.plugins.arg_packing module ml_switcheroo.plugins.attention_packing module ml_switcheroo.plugins.batch_norm module ml_switcheroo.plugins.casting module ml_switcheroo.plugins.checkpoint_keys module ml_switcheroo.plugins.clipping module ml_switcheroo.plugins.context_to_function_wrap module ml_switcheroo.plugins.data_loader module ml_switcheroo.plugins.decompositions module ml_switcheroo.plugins.device_allocator module ml_switcheroo.plugins.device_checks module ml_switcheroo.plugins.einsum module ml_switcheroo.plugins.flatten module ml_switcheroo.plugins.gather module ml_switcheroo.plugins.inplace_unroll module ml_switcheroo.plugins.io_handler module ml_switcheroo.plugins.keras_sequential module ml_switcheroo.plugins.loop_unroll module ml_switcheroo.plugins.loss_wrapper module ml_switcheroo.plugins.method_property module ml_switcheroo.plugins.mlx_extras module

ml_switcheroo.plugins.mlx_optimizers module ml_switcheroo.plugins.nnx_to_torch_params module ml_switcheroo.plugins.optimizer_step module ml_switcheroo.plugins.padding module ml_switcheroo.plugins.reshape module ml_switcheroo.plugins.rng_threading module ml_switcheroo.plugins.scatter module ml_switcheroo.plugins.schedulers module ml_switcheroo.plugins.shape_packing module ml_switcheroo.plugins.state_container module ml_switcheroo.plugins.state_flag_injection module ml_switcheroo.plugins.static_unroll module ml_switcheroo.plugins.tf_data_loader module ml_switcheroo.plugins.topk module ml_switcheroo.semantics module ml_switcheroo.semantics.autogen module ml_switcheroo.semantics.manager module ml_switcheroo.semantics.merging module ml_switcheroo.semantics.paths module ml_switcheroo.semantics.schema module ml_switcheroo.semantics.standards_internal module ml_switcheroo.testing module ml_switcheroo.testing.batch_runner module ml_switcheroo.testing.fuzzer module ml_switcheroo.testing.harness_generator module ml_switcheroo.testing.harness_generator_template module ml_switcheroo.testing.linter module ml_switcheroo.testing.runner module ml_switcheroo.utils module ml_switcheroo.utils.code_extractor module ml_switcheroo.utils.console module ml_switcheroo.utils.doc_gen module ml_switcheroo.utils.node_diff module ml_switcheroo.utils.readme_editor module MLXAdapter (class in ml_switcheroo.frameworks.mlx) (in module ml_switcheroo.frameworks) model_config (ml_switcheroo.core.dsl.Rule attribute) (ml_switcheroo.core.hooks.AutoWireSpec attribute) (ml_switcheroo.semantics.schema.FrameworkTraits attribute) (ml_switcheroo.semantics.schema.OpDefinition attribute) (ml_switcheroo.semantics.schema.Rule attribute) (ml_switcheroo.semantics.schema.SemanticsFile attribute) (ml_switcheroo.semantics.schema.StructuralTraits attribute) (ml_switcheroo.semantics.schema.Variant attribute) module injector_fw injector_plugin injector_spec ml_switcheroo ml_switcheroo.__main__ ml_switcheroo.analysis ml_switcheroo.analysis.audit ml_switcheroo.analysis.dependencies ml_switcheroo.analysis.lifecycle ml_switcheroo.analysis.purity ml_switcheroo.cli ml_switcheroo.cli.__main__ ml_switcheroo.cli.commands ml_switcheroo.cli.handlers ml_switcheroo.cli.handlers.audit ml_switcheroo.cli.handlers.convert ml_switcheroo.cli.handlers.define ml_switcheroo.cli.handlers.dev ml_switcheroo.cli.handlers.discovery ml_switcheroo.cli.handlers.learning ml_switcheroo.cli.handlers.snapshots ml_switcheroo.cli.handlers.verify ml_switcheroo.cli.matrix ml_switcheroo.cli.wizard ml_switcheroo.config ml_switcheroo.core ml_switcheroo.core.discovery ml_switcheroo.core.dsl ml_switcheroo.core.engine ml_switcheroo.core.escape_hatch ml_switcheroo.core.ghost ml_switcheroo.core.hooks ml_switcheroo.core.import_fixer ml_switcheroo.core.rewriter ml_switcheroo.core.rewriter.attributes ml_switcheroo.core.rewriter.base ml_switcheroo.core.rewriter.calls ml_switcheroo.core.rewriter.calls.dispatch ml_switcheroo.core.rewriter.calls.mixer ml_switcheroo.core.rewriter.calls.transformers ml_switcheroo.core.rewriter.calls.utils ml_switcheroo.core.rewriter.contexts ml_switcheroo.core.rewriter.control_flow ml_switcheroo.core.rewriter.decorators ml_switcheroo.core.rewriter.initializers ml_switcheroo.core.rewriter.normalization ml_switcheroo.core.rewriter.structure ml_switcheroo.core.rewriter.structure_class ml_switcheroo.core.rewriter.structure_func ml_switcheroo.core.rewriter.structure_types ml_switcheroo.core.rewriter.types ml_switcheroo.core.scanners ml_switcheroo.core.tracer ml_switcheroo.discovery ml_switcheroo.discovery.consensus ml_switcheroo.discovery.harvester ml_switcheroo.discovery.inspector ml_switcheroo.discovery.scaffolder ml_switcheroo.discovery.syncer ml_switcheroo.enums ml_switcheroo.frameworks ml_switcheroo.frameworks.base ml_switcheroo.frameworks.common ml_switcheroo.frameworks.common.data ml_switcheroo.frameworks.common.jax_stack ml_switcheroo.frameworks.flax_nnx ml_switcheroo.frameworks.jax ml_switcheroo.frameworks.keras ml_switcheroo.frameworks.mlx ml_switcheroo.frameworks.numpy ml_switcheroo.frameworks.optax_shim ml_switcheroo.frameworks.paxml ml_switcheroo.frameworks.tensorflow ml_switcheroo.frameworks.torch ml_switcheroo.generated_tests ml_switcheroo.generated_tests.generator ml_switcheroo.importers ml_switcheroo.importers.array_api_reader ml_switcheroo.importers.onnx_reader ml_switcheroo.plugins ml_switcheroo.plugins.arg_packing ml_switcheroo.plugins.attention_packing ml_switcheroo.plugins.batch_norm ml_switcheroo.plugins.casting ml_switcheroo.plugins.checkpoint_keys ml_switcheroo.plugins.clipping ml_switcheroo.plugins.context_to_function_wrap ml_switcheroo.plugins.data_loader ml_switcheroo.plugins.decompositions ml_switcheroo.plugins.device_allocator ml_switcheroo.plugins.device_checks ml_switcheroo.plugins.einsum ml_switcheroo.plugins.flatten ml_switcheroo.plugins.gather ml_switcheroo.plugins.inplace_unroll ml_switcheroo.plugins.io_handler ml_switcheroo.plugins.keras_sequential ml_switcheroo.plugins.loop_unroll ml_switcheroo.plugins.loss_wrapper ml_switcheroo.plugins.method_property ml_switcheroo.plugins.mlx_extras ml_switcheroo.plugins.mlx_optimizers ml_switcheroo.plugins.nnx_to_torch_params ml_switcheroo.plugins.optimizer_step ml_switcheroo.plugins.padding ml_switcheroo.plugins.reshape ml_switcheroo.plugins.rng_threading ml_switcheroo.plugins.scatter ml_switcheroo.plugins.schedulers ml_switcheroo.plugins.shape_packing ml_switcheroo.plugins.state_container ml_switcheroo.plugins.state_flag_injection ml_switcheroo.plugins.static_unroll ml_switcheroo.plugins.tf_data_loader ml_switcheroo.plugins.topk ml_switcheroo.semantics ml_switcheroo.semantics.autogen ml_switcheroo.semantics.manager ml_switcheroo.semantics.merging ml_switcheroo.semantics.paths ml_switcheroo.semantics.schema ml_switcheroo.semantics.standards_internal ml_switcheroo.testing ml_switcheroo.testing.batch_runner ml_switcheroo.testing.fuzzer ml_switcheroo.testing.harness_generator ml_switcheroo.testing.harness_generator_template ml_switcheroo.testing.linter ml_switcheroo.testing.runner ml_switcheroo.utils ml_switcheroo.utils.code_extractor ml_switcheroo.utils.console ml_switcheroo.utils.doc_gen ml_switcheroo.utils.node_diff ml_switcheroo.utils.readme_editor module (ml_switcheroo.core.dsl.ImportReq attribute) (ml_switcheroo.semantics.schema.FrameworkAlias attribute) module_base (ml_switcheroo.semantics.schema.StructuralTraits attribute)

N

name (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.dsl.PluginScaffoldDef attribute) (ml_switcheroo.core.ghost.GhostParam attribute) (ml_switcheroo.core.ghost.GhostRef attribute) (ml_switcheroo.discovery.consensus.CandidateStandard attribute) (ml_switcheroo.semantics.schema.FrameworkAlias attribute) NEQ (ml_switcheroo.enums.LogicOp attribute) NEURAL (ml_switcheroo.enums.SemanticTier attribute), [1] nondeterministic (ml_switcheroo.core.dsl.OperationDef attribute) NormalizationMixin (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.normalization)

normalize_arg() (ml_switcheroo.cli.commands.ConsensusEngine class method) (ml_switcheroo.discovery.consensus.ConsensusEngine class method) normalize_einsum() (in module ml_switcheroo.plugins.einsum) normalize_harness_imports() (ml_switcheroo.utils.code_extractor.CodeExtractor static method) normalize_name() (ml_switcheroo.cli.commands.ConsensusEngine class method) (ml_switcheroo.discovery.consensus.ConsensusEngine class method) NOT_IN (ml_switcheroo.enums.LogicOp attribute) NumpyAdapter (class in ml_switcheroo.frameworks.numpy) (in module ml_switcheroo.frameworks)

O

OnnxSpecImporter (class in ml_switcheroo.importers.onnx_reader) op (ml_switcheroo.core.dsl.Rule attribute) (ml_switcheroo.semantics.schema.Rule attribute) op_def (injector_spec.StandardsInjector attribute), [1] op_name (injector_fw.FrameworkInjector attribute), [1] op_type (ml_switcheroo.core.dsl.OperationDef attribute) OpDefinition (class in ml_switcheroo.semantics.schema) operation (ml_switcheroo.core.dsl.OperationDef attribute) OperationDef (class in ml_switcheroo.core.dsl) operator (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) (ml_switcheroo.semantics.schema.Variant attribute)

ops (ml_switcheroo.core.hooks.AutoWireSpec attribute) optax (in module ml_switcheroo.frameworks.optax_shim) OptaxScanner (class in ml_switcheroo.frameworks.optax_shim) OPTIMIZER (ml_switcheroo.frameworks.base.StandardCategory attribute) options (ml_switcheroo.core.dsl.ParameterDef attribute) OpType (class in ml_switcheroo.core.dsl) output_adapter (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) (ml_switcheroo.semantics.schema.Variant attribute) output_cast (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) output_select_index (ml_switcheroo.core.dsl.FrameworkVariant attribute) output_shape_calc (ml_switcheroo.core.dsl.OperationDef attribute)

P

pack_to_tuple (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) pack_varargs() (in module ml_switcheroo.plugins.arg_packing) ParameterDef (class in ml_switcheroo.core.dsl) params (ml_switcheroo.core.ghost.GhostRef attribute) parent_id (ml_switcheroo.core.tracer.TraceEvent attribute) parse() (ml_switcheroo.ASTEngine method) (ml_switcheroo.core.engine.ASTEngine method) parse_cli_key_values() (in module ml_switcheroo.config) parse_file() (ml_switcheroo.importers.onnx_reader.OnnxSpecImporter method) parse_folder() (ml_switcheroo.importers.array_api_reader.ArrayApiSpecImporter method) parse_plugin_settings() (ml_switcheroo.config.RuntimeConfig method) (ml_switcheroo.RuntimeConfig method) PaxmlAdapter (class in ml_switcheroo.frameworks.paxml) (in module ml_switcheroo.frameworks) persist() (ml_switcheroo.cli.commands.SemanticPersister method) (ml_switcheroo.semantics.autogen.SemanticPersister method)

PHASE_END (ml_switcheroo.core.tracer.TraceEventType attribute) PHASE_START (ml_switcheroo.core.tracer.TraceEventType attribute) PivotRewriter (class in ml_switcheroo.core.rewriter) plugin_paths (ml_switcheroo.config.RuntimeConfig attribute) (ml_switcheroo.RuntimeConfig attribute) plugin_settings (ml_switcheroo.config.RuntimeConfig attribute) (ml_switcheroo.RuntimeConfig attribute) PluginGenerator (class in injector_plugin) plugins_dir (injector_plugin.PluginGenerator attribute), [1] PluginScaffoldDef (class in ml_switcheroo.core.dsl) PluginType (class in ml_switcheroo.core.dsl) praxis (in module ml_switcheroo.frameworks.paxml) preamble_stmts (ml_switcheroo.core.rewriter.types.SignatureContext attribute) PreambleInjectorType (in module ml_switcheroo.core.hooks) preserve_source (ml_switcheroo.core.import_fixer.ImportFixer attribute) PurityScanner (class in ml_switcheroo.analysis.purity)

R

rank (ml_switcheroo.core.dsl.ParameterDef attribute) raw_config() (ml_switcheroo.core.hooks.HookContext method) readme_path (ml_switcheroo.cli.commands.ReadmeEditor attribute), [1] (ml_switcheroo.utils.readme_editor.ReadmeEditor attribute), [1] ReadmeEditor (class in ml_switcheroo.cli.commands) (class in ml_switcheroo.utils.readme_editor) register_framework() (in module ml_switcheroo.frameworks) (in module ml_switcheroo.frameworks.base) register_hook() (in module ml_switcheroo.core.hooks) render() (ml_switcheroo.cli.matrix.CompatibilityMatrix method) repack_attention() (in module ml_switcheroo.plugins.attention_packing) replaced_by (ml_switcheroo.core.dsl.OperationDef attribute) required_imports (ml_switcheroo.core.dsl.FrameworkVariant attribute) requires_plugin (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) (ml_switcheroo.semantics.schema.Variant attribute) requires_super_init (ml_switcheroo.semantics.schema.StructuralTraits attribute) reset_console() (in module ml_switcheroo.utils.console) reset_tracer() (in module ml_switcheroo.core.tracer) resolve_roots() (in module ml_switcheroo.cli.handlers.audit) resolve_semantics_dir() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.semantics.paths) resolve_snapshots_dir() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.semantics.paths) resolve_variant() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method)

results (ml_switcheroo.analysis.audit.CoverageScanner attribute), [1] return_type (ml_switcheroo.core.dsl.OperationDef attribute) rewrite_as_infix() (in module ml_switcheroo.core.rewriter.calls.transformers) rewrite_as_inline_lambda() (in module ml_switcheroo.core.rewriter.calls.transformers) rewrite_as_macro() (in module ml_switcheroo.core.rewriter.calls.transformers) rewrite_stateful_call() (in module ml_switcheroo.core.rewriter.calls.utils) rng_seed_methods (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) root_fw (ml_switcheroo.discovery.harvester.ImportScanner attribute) Rule (class in ml_switcheroo.core.dsl) (class in ml_switcheroo.semantics.schema) rules (ml_switcheroo.core.dsl.PluginScaffoldDef attribute) run() (ml_switcheroo.ASTEngine method) (ml_switcheroo.core.engine.ASTEngine method) run_all() (ml_switcheroo.cli.commands.BatchValidator method) (ml_switcheroo.testing.batch_runner.BatchValidator method) runner (ml_switcheroo.cli.commands.BatchValidator attribute) (ml_switcheroo.testing.batch_runner.BatchValidator attribute) RuntimeConfig (class in ml_switcheroo) (class in ml_switcheroo.config)

S

scaffold() (ml_switcheroo.discovery.scaffolder.Scaffolder method) scaffold_plugins (ml_switcheroo.core.dsl.OperationDef attribute) Scaffolder (class in ml_switcheroo.discovery.scaffolder) scan_losses() (ml_switcheroo.frameworks.optax_shim.OptaxScanner static method) scan_optimizers() (ml_switcheroo.frameworks.optax_shim.OptaxScanner static method) score (ml_switcheroo.discovery.consensus.CandidateStandard attribute) search_modules (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) SemanticHarvester (class in ml_switcheroo.discovery.harvester) SemanticPersister (class in ml_switcheroo.cli.commands) (class in ml_switcheroo.semantics.autogen) semantics (ml_switcheroo.analysis.audit.CoverageScanner attribute) (ml_switcheroo.analysis.dependencies.DependencyScanner attribute) (ml_switcheroo.ASTEngine attribute) (ml_switcheroo.cli.commands.BatchValidator attribute) (ml_switcheroo.cli.commands.ReadmeEditor attribute), [1] (ml_switcheroo.cli.matrix.CompatibilityMatrix attribute) (ml_switcheroo.cli.wizard.MappingWizard attribute) (ml_switcheroo.core.engine.ASTEngine attribute) (ml_switcheroo.core.hooks.HookContext attribute) (ml_switcheroo.core.rewriter.base.BaseRewriter attribute) (ml_switcheroo.core.rewriter.BaseRewriter attribute) (ml_switcheroo.discovery.harvester.SemanticHarvester attribute) (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) (ml_switcheroo.testing.batch_runner.BatchValidator attribute) (ml_switcheroo.utils.doc_gen.MigrationGuideGenerator attribute), [1] (ml_switcheroo.utils.readme_editor.ReadmeEditor attribute), [1] semantics_mgr (ml_switcheroo.generated_tests.generator.TestGenerator attribute), [1] SemanticsFile (class in ml_switcheroo.semantics.schema) SemanticsManager (class in ml_switcheroo) (class in ml_switcheroo.cli.commands) (class in ml_switcheroo.semantics.manager) SemanticsManagerType (in module ml_switcheroo.core.hooks) SemanticTier (class in ml_switcheroo.enums) set_console() (in module ml_switcheroo.utils.console) shape_spec (ml_switcheroo.core.dsl.ParameterDef attribute) should_traverse() (ml_switcheroo.core.scanners.SimpleNameScanner method) SignatureContext (class in ml_switcheroo.core.rewriter.types) similarity_threshold (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) SimpleNameScanner (class in ml_switcheroo.core.scanners) SimulatedReflection (class in ml_switcheroo.core.discovery) skip_fuzzing (ml_switcheroo.core.dsl.OperationDef attribute) SNAPSHOT_DIR (in module ml_switcheroo.frameworks.base) source (ml_switcheroo.ASTEngine attribute) (ml_switcheroo.core.engine.ASTEngine attribute) source_flavour (ml_switcheroo.config.RuntimeConfig attribute) (ml_switcheroo.RuntimeConfig attribute) source_framework (ml_switcheroo.config.RuntimeConfig attribute) (ml_switcheroo.RuntimeConfig attribute)

source_fw (ml_switcheroo.analysis.dependencies.DependencyScanner attribute) (ml_switcheroo.analysis.purity.PurityScanner attribute) (ml_switcheroo.core.hooks.HookContext attribute) (ml_switcheroo.core.rewriter.base.BaseRewriter attribute) (ml_switcheroo.core.rewriter.BaseRewriter attribute) (ml_switcheroo.core.scanners.UsageScanner attribute) staged_mappings (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) staged_specs (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) StandardCategory (class in ml_switcheroo.frameworks.base) StandardMap (class in ml_switcheroo.frameworks.base) StandardsInjector (class in injector_spec) start() (ml_switcheroo.cli.wizard.MappingWizard method) START_MARKER (ml_switcheroo.core.escape_hatch.EscapeHatch attribute) start_phase() (ml_switcheroo.core.tracer.TraceLogger method) stateful_call (ml_switcheroo.semantics.schema.FrameworkTraits attribute) std_args (ml_switcheroo.core.dsl.OperationDef attribute) (ml_switcheroo.discovery.consensus.CandidateStandard attribute) (ml_switcheroo.semantics.schema.OpDefinition attribute) std_args_info (ml_switcheroo.discovery.harvester.TargetCallVisitor attribute) strict_mode (ml_switcheroo.ASTEngine attribute) (ml_switcheroo.config.RuntimeConfig attribute) (ml_switcheroo.core.engine.ASTEngine attribute) (ml_switcheroo.core.rewriter.base.BaseRewriter attribute) (ml_switcheroo.core.rewriter.BaseRewriter attribute) (ml_switcheroo.RuntimeConfig attribute) strip_kwarg() (in module ml_switcheroo.core.rewriter.calls.utils) strip_magic_args (ml_switcheroo.semantics.schema.StructuralTraits attribute) strip_zero_grad() (in module ml_switcheroo.plugins.optimizer_step) structural_traits (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) StructuralLinter (class in ml_switcheroo.testing.linter) StructuralTraits (class in ml_switcheroo.semantics.schema) StructureMixin (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.structure) submodule_map (ml_switcheroo.core.import_fixer.ImportFixer attribute) success (ml_switcheroo.ConversionResult attribute) (ml_switcheroo.core.engine.ConversionResult attribute) SUCCESS_LEVEL_NUM (in module ml_switcheroo.utils.console) supported_tiers (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) sync() (ml_switcheroo.cli.commands.FrameworkSyncer method) (ml_switcheroo.discovery.syncer.FrameworkSyncer method)

T

T (in module ml_switcheroo.config) (in module ml_switcheroo.core.hooks) target (ml_switcheroo.ASTEngine attribute) (ml_switcheroo.core.engine.ASTEngine attribute) target_api (ml_switcheroo.discovery.harvester.TargetCallVisitor attribute) target_flavour (ml_switcheroo.config.RuntimeConfig attribute) (ml_switcheroo.RuntimeConfig attribute) target_framework (ml_switcheroo.config.RuntimeConfig attribute) (ml_switcheroo.RuntimeConfig attribute) target_fw (injector_fw.FrameworkInjector attribute), [1] (ml_switcheroo.core.hooks.HookContext attribute) (ml_switcheroo.core.import_fixer.ImportFixer attribute) (ml_switcheroo.core.rewriter.base.BaseRewriter attribute) (ml_switcheroo.core.rewriter.BaseRewriter attribute) (ml_switcheroo.discovery.harvester.SemanticHarvester attribute) target_name (ml_switcheroo.core.scanners.SimpleNameScanner attribute) TargetCallVisitor (class in ml_switcheroo.discovery.harvester) TEMPLATE_FUNC_DEF (in module injector_plugin) TEMPLATE_HEADER (in module injector_plugin) TensorFlowAdapter (class in ml_switcheroo.frameworks.tensorflow) (in module ml_switcheroo.frameworks) test_atol (ml_switcheroo.core.dsl.OperationDef attribute) test_config (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) test_rtol (ml_switcheroo.core.dsl.OperationDef attribute) test_templates (ml_switcheroo.cli.commands.SemanticsManager attribute) (ml_switcheroo.semantics.manager.SemanticsManager attribute) (ml_switcheroo.SemanticsManager attribute) TestGenerator (class in ml_switcheroo.generated_tests.generator) tiers (ml_switcheroo.semantics.schema.FrameworkTraits attribute) timestamp (ml_switcheroo.core.tracer.TraceEvent attribute) to_source() (ml_switcheroo.ASTEngine method) (ml_switcheroo.core.engine.ASTEngine method) tomllib (in module ml_switcheroo.config) torch (in module ml_switcheroo.frameworks.torch) TorchAdapter (class in ml_switcheroo.frameworks.torch) (in module ml_switcheroo.frameworks) trace_events (ml_switcheroo.ConversionResult attribute) (ml_switcheroo.core.engine.ConversionResult attribute)

TraceEvent (class in ml_switcheroo.core.tracer) TraceEventType (class in ml_switcheroo.core.tracer) TraceLogger (class in ml_switcheroo.core.tracer) traits (ml_switcheroo.semantics.schema.FrameworkTraits attribute) transform_alpha_add() (in module ml_switcheroo.plugins.decompositions) transform_alpha_add_reverse() (in module ml_switcheroo.plugins.decompositions) transform_batch_norm() (in module ml_switcheroo.plugins.batch_norm) transform_casting() (in module ml_switcheroo.plugins.casting) transform_checkpoint_keys() (in module ml_switcheroo.plugins.checkpoint_keys) transform_compiler() (in module ml_switcheroo.plugins.mlx_extras) transform_context_manager() (in module ml_switcheroo.plugins.context_to_function_wrap) transform_cuda_check() (in module ml_switcheroo.plugins.device_checks) transform_dataloader() (in module ml_switcheroo.plugins.data_loader) transform_device_allocator() (in module ml_switcheroo.plugins.device_allocator) transform_flatten() (in module ml_switcheroo.plugins.flatten) transform_gather() (in module ml_switcheroo.plugins.gather) transform_grad_clipping() (in module ml_switcheroo.plugins.clipping) transform_io_calls() (in module ml_switcheroo.plugins.io_handler) transform_keras_sequential() (in module ml_switcheroo.plugins.keras_sequential) transform_loops() (in module ml_switcheroo.plugins.loop_unroll) transform_loss_reduction() (in module ml_switcheroo.plugins.loss_wrapper) transform_method_to_property() (in module ml_switcheroo.plugins.method_property) transform_mlx_optimizer_init() (in module ml_switcheroo.plugins.mlx_optimizers) transform_mlx_optimizer_step() (in module ml_switcheroo.plugins.mlx_optimizers) transform_mlx_zero_grad() (in module ml_switcheroo.plugins.mlx_optimizers) transform_nnx_param() (in module ml_switcheroo.plugins.nnx_to_torch_params) transform_optimizer_init() (in module ml_switcheroo.plugins.optimizer_step) transform_optimizer_step() (in module ml_switcheroo.plugins.optimizer_step) transform_padding() (in module ml_switcheroo.plugins.padding) transform_scatter() (in module ml_switcheroo.plugins.scatter) transform_scheduler_init() (in module ml_switcheroo.plugins.schedulers) transform_scheduler_step() (in module ml_switcheroo.plugins.schedulers) transform_shape_packing() (in module ml_switcheroo.plugins.shape_packing) transform_synchronize() (in module ml_switcheroo.plugins.mlx_extras) transform_tf_dataloader() (in module ml_switcheroo.plugins.tf_data_loader) transform_topk() (in module ml_switcheroo.plugins.topk) transform_view_semantics() (in module ml_switcheroo.plugins.reshape) transformation_type (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) (ml_switcheroo.semantics.schema.Variant attribute) type (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.dsl.PluginScaffoldDef attribute) (ml_switcheroo.core.tracer.TraceEvent attribute) TYPE_MAP (in module ml_switcheroo.plugins.casting) TypeStructureMixin (class in ml_switcheroo.core.rewriter.structure_types)

U

ui_priority (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter attribute) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter attribute) (ml_switcheroo.frameworks.keras.KerasAdapter attribute) (ml_switcheroo.frameworks.mlx.MLXAdapter attribute) (ml_switcheroo.frameworks.numpy.NumpyAdapter attribute) (ml_switcheroo.frameworks.paxml.PaxmlAdapter attribute) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter attribute) (ml_switcheroo.frameworks.torch.TorchAdapter attribute)

unknown_imports (ml_switcheroo.analysis.dependencies.DependencyScanner attribute) unroll_inplace_ops() (in module ml_switcheroo.plugins.inplace_unroll) unroll_static_loops() (in module ml_switcheroo.plugins.static_unroll) update_definition() (ml_switcheroo.cli.commands.SemanticsManager method) (ml_switcheroo.semantics.manager.SemanticsManager method) (ml_switcheroo.SemanticsManager method) update_matrix() (ml_switcheroo.cli.commands.ReadmeEditor method) (ml_switcheroo.utils.readme_editor.ReadmeEditor method) UsageScanner (class in ml_switcheroo.core.scanners) use_api (ml_switcheroo.core.dsl.Rule attribute) (ml_switcheroo.semantics.schema.Rule attribute)

V

validate_framework() (ml_switcheroo.config.RuntimeConfig class method) (ml_switcheroo.RuntimeConfig class method) validate_settings() (ml_switcheroo.core.hooks.HookContext method) validate_transpilation() (in module ml_switcheroo.testing.linter) validation_report (ml_switcheroo.config.RuntimeConfig attribute) (ml_switcheroo.RuntimeConfig attribute) value (ml_switcheroo.plugins.static_unroll.LoopVarReplacer attribute), [1] var_name (ml_switcheroo.plugins.static_unroll.LoopVarReplacer attribute), [1] Variant (class in ml_switcheroo.semantics.schema) variant (injector_fw.FrameworkInjector attribute), [1] variants (ml_switcheroo.core.dsl.OperationDef attribute) (ml_switcheroo.discovery.consensus.CandidateStandard attribute) (ml_switcheroo.semantics.schema.OpDefinition attribute) verify() (ml_switcheroo.testing.runner.EquivalenceRunner method) violations (ml_switcheroo.testing.linter.StructuralLinter attribute), [1] visit_AnnAssign() (ml_switcheroo.analysis.lifecycle.InitializationTracker method) visit_Annotation() (ml_switcheroo.core.rewriter.structure_types.TypeStructureMixin method) visit_Assign() (ml_switcheroo.analysis.lifecycle.InitializationTracker method) visit_Attribute() (ml_switcheroo.analysis.audit.CoverageScanner method) (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.testing.linter.StructuralLinter method) visit_Call() (ml_switcheroo.analysis.audit.CoverageScanner method) (ml_switcheroo.analysis.purity.PurityScanner method) (ml_switcheroo.discovery.harvester.TargetCallVisitor method) visit_ClassDef() (injector_fw.FrameworkInjector method) (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.core.rewriter.structure_class.ClassStructureMixin method) visit_FunctionDef() (injector_fw.FrameworkInjector method) (injector_plugin.BodyExtractor method) (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.core.rewriter.structure_func.FuncStructureMixin method)

visit_Global() (ml_switcheroo.analysis.purity.PurityScanner method) visit_Import() (injector_fw.FrameworkInjector method) (ml_switcheroo.analysis.audit.CoverageScanner method) (ml_switcheroo.analysis.dependencies.DependencyScanner method) (ml_switcheroo.core.rewriter.base.BaseRewriter method) (ml_switcheroo.core.rewriter.BaseRewriter method) (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.discovery.harvester.ImportScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) visit_ImportFrom() (injector_fw.FrameworkInjector method) (ml_switcheroo.analysis.audit.CoverageScanner method) (ml_switcheroo.analysis.dependencies.DependencyScanner method) (ml_switcheroo.core.rewriter.base.BaseRewriter method) (ml_switcheroo.core.rewriter.BaseRewriter method) (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.discovery.harvester.ImportScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) visit_Name() (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) visit_Nonlocal() (ml_switcheroo.analysis.purity.PurityScanner method) visit_SimpleStatementLine() (ml_switcheroo.analysis.purity.PurityScanner method) (ml_switcheroo.core.rewriter.base.BaseRewriter method) (ml_switcheroo.core.rewriter.BaseRewriter method)

W

warnings (ml_switcheroo.analysis.lifecycle.InitializationTracker attribute), [1]

Y

yaml (in module ml_switcheroo.cli.handlers.define)