Python Module Index

d | g | i | m | p | s | t | u
 
d
dispatch
 
g
guards
 
i
injector_fw
    injector_fw.core
    injector_fw.utils
injector_plugin
injector_spec
 
m
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.analysis.symbol_table
    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.convert_weights
    ml_switcheroo.cli.handlers.define
    ml_switcheroo.cli.handlers.dev
    ml_switcheroo.cli.handlers.discovery
    ml_switcheroo.cli.handlers.learning
    ml_switcheroo.cli.handlers.meta
    ml_switcheroo.cli.handlers.snapshots
    ml_switcheroo.cli.handlers.suggest
    ml_switcheroo.cli.handlers.verify
    ml_switcheroo.cli.matrix
    ml_switcheroo.cli.wizard
    ml_switcheroo.compiler
    ml_switcheroo.compiler.backend
    ml_switcheroo.compiler.backends
    ml_switcheroo.compiler.backends.extras
    ml_switcheroo.compiler.backends.python
    ml_switcheroo.compiler.backends.python_snippet
    ml_switcheroo.compiler.backends.rdna
    ml_switcheroo.compiler.backends.rdna.emitter
    ml_switcheroo.compiler.backends.rdna.macros
    ml_switcheroo.compiler.backends.rdna.synthesizer
    ml_switcheroo.compiler.backends.sass
    ml_switcheroo.compiler.backends.sass.emitter
    ml_switcheroo.compiler.backends.sass.macros
    ml_switcheroo.compiler.backends.sass.synthesizer
    ml_switcheroo.compiler.differ
    ml_switcheroo.compiler.frontends
    ml_switcheroo.compiler.frontends.python
    ml_switcheroo.compiler.frontends.rdna
    ml_switcheroo.compiler.frontends.rdna.analysis
    ml_switcheroo.compiler.frontends.rdna.lifter
    ml_switcheroo.compiler.frontends.rdna.nodes
    ml_switcheroo.compiler.frontends.rdna.parser
    ml_switcheroo.compiler.frontends.rdna.tokens
    ml_switcheroo.compiler.frontends.sass
    ml_switcheroo.compiler.frontends.sass.analysis
    ml_switcheroo.compiler.frontends.sass.lifter
    ml_switcheroo.compiler.frontends.sass.nodes
    ml_switcheroo.compiler.frontends.sass.parser
    ml_switcheroo.compiler.frontends.sass.tokens
    ml_switcheroo.compiler.ir
    ml_switcheroo.compiler.registry
    ml_switcheroo.config
    ml_switcheroo.core
    ml_switcheroo.core.conversion_result
    ml_switcheroo.core.discovery
    ml_switcheroo.core.dsl
    ml_switcheroo.core.engine
    ml_switcheroo.core.escape_hatch
    ml_switcheroo.core.ghost
    ml_switcheroo.core.graph
    ml_switcheroo.core.graph_optimizer
    ml_switcheroo.core.hooks
    ml_switcheroo.core.html
    ml_switcheroo.core.html.nodes
    ml_switcheroo.core.html.parser
    ml_switcheroo.core.import_fixer
    ml_switcheroo.core.import_fixer.attributes_mixin
    ml_switcheroo.core.import_fixer.base
    ml_switcheroo.core.import_fixer.imports_mixin
    ml_switcheroo.core.import_fixer.injection_mixin
    ml_switcheroo.core.import_fixer.resolution
    ml_switcheroo.core.import_fixer.utils
    ml_switcheroo.core.ingestion
    ml_switcheroo.core.latex
    ml_switcheroo.core.latex.nodes
    ml_switcheroo.core.latex.parser
    ml_switcheroo.core.mlir
    ml_switcheroo.core.mlir.dialect
    ml_switcheroo.core.mlir.emitter
    ml_switcheroo.core.mlir.gen_base
    ml_switcheroo.core.mlir.gen_expressions
    ml_switcheroo.core.mlir.gen_statements
    ml_switcheroo.core.mlir.generator
    ml_switcheroo.core.mlir.naming
    ml_switcheroo.core.mlir.nodes
    ml_switcheroo.core.mlir.parser
    ml_switcheroo.core.mlir.stablehlo_emitter
    ml_switcheroo.core.mlir.tokens
    ml_switcheroo.core.rewriter
    ml_switcheroo.core.rewriter.context
    ml_switcheroo.core.rewriter.interface
    ml_switcheroo.core.rewriter.normalization_utils
    ml_switcheroo.core.rewriter.passes
    ml_switcheroo.core.rewriter.passes.api
    ml_switcheroo.core.rewriter.passes.auxiliary
    ml_switcheroo.core.rewriter.passes.structure
    ml_switcheroo.core.rewriter.patcher
    ml_switcheroo.core.rewriter.pipeline
    ml_switcheroo.core.rewriter.types
    ml_switcheroo.core.scanners
    ml_switcheroo.core.tikz
    ml_switcheroo.core.tikz.analyser
    ml_switcheroo.core.tikz.nodes
    ml_switcheroo.core.tikz.parser
    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.jax_stack
    ml_switcheroo.frameworks.flax_nnx
    ml_switcheroo.frameworks.html_dsl
    ml_switcheroo.frameworks.jax
    ml_switcheroo.frameworks.keras
    ml_switcheroo.frameworks.latex_dsl
    ml_switcheroo.frameworks.loader
    ml_switcheroo.frameworks.mlir
    ml_switcheroo.frameworks.mlx
    ml_switcheroo.frameworks.numpy
    ml_switcheroo.frameworks.optax_shim
    ml_switcheroo.frameworks.paxml
    ml_switcheroo.frameworks.rdna
    ml_switcheroo.frameworks.sass
    ml_switcheroo.frameworks.stablehlo
    ml_switcheroo.frameworks.tensorflow
    ml_switcheroo.frameworks.tikz
    ml_switcheroo.frameworks.torch
    ml_switcheroo.generated_tests
    ml_switcheroo.generated_tests.generator
    ml_switcheroo.generated_tests.inputs
    ml_switcheroo.generated_tests.runtime
    ml_switcheroo.generated_tests.runtime_builder
    ml_switcheroo.generated_tests.templates
    ml_switcheroo.importers
    ml_switcheroo.importers.array_api_reader
    ml_switcheroo.importers.onnx_reader
    ml_switcheroo.importers.sass_reader
    ml_switcheroo.importers.stablehlo_reader
    ml_switcheroo.plugins
    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.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.plugins.utils
    ml_switcheroo.semantics
    ml_switcheroo.semantics.autogen
    ml_switcheroo.semantics.file_loader
    ml_switcheroo.semantics.manager
    ml_switcheroo.semantics.merging
    ml_switcheroo.semantics.paths
    ml_switcheroo.semantics.registry_loader
    ml_switcheroo.semantics.schema
    ml_switcheroo.testing
    ml_switcheroo.testing.batch_runner
    ml_switcheroo.testing.bisector
    ml_switcheroo.testing.fuzzer
    ml_switcheroo.testing.fuzzer.core
    ml_switcheroo.testing.fuzzer.generators
    ml_switcheroo.testing.fuzzer.heuristics
    ml_switcheroo.testing.fuzzer.parser
    ml_switcheroo.testing.fuzzer.strategies
    ml_switcheroo.testing.fuzzer.utils
    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_context
    ml_switcheroo.utils.doc_gen
    ml_switcheroo.utils.doc_renderer
    ml_switcheroo.utils.node_diff
    ml_switcheroo.utils.readme_editor
    ml_switcheroo.utils.visualizer
 
p
post
pre
 
s
strategy
 
t
transformers
 
u
utils