API Reference¶
This page contains auto-generated API reference documentation [1].
- injector_fw
- ml_switcheroo
- ml_switcheroo.__main__
- ml_switcheroo.analysis
- ml_switcheroo.cli
- 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.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.enums
- ml_switcheroo.frameworks
- ml_switcheroo.frameworks.base
- ml_switcheroo.frameworks.common
- 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.importers
- 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.testing
- ml_switcheroo.utils
- injector_spec
- injector_plugin