API Reference¶
This page contains auto-generated API reference documentation [1].
- pre
- post
- utils
- guards
- dispatch
- strategy
- injector_fw
- transformers
- ml_switcheroo
- ml_switcheroo.__main__
- ml_switcheroo.analysis
- 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.differ
- ml_switcheroo.compiler.frontends
- 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.import_fixer
- ml_switcheroo.core.ingestion
- ml_switcheroo.core.latex
- 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.scanners
- ml_switcheroo.core.tikz
- ml_switcheroo.core.tracer
- ml_switcheroo.discovery
- ml_switcheroo.enums
- ml_switcheroo.frameworks
- ml_switcheroo.frameworks.base
- ml_switcheroo.frameworks.common
- 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.importers
- 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.testing
- ml_switcheroo.utils
- injector_spec
- injector_plugin