Python Module Index

i | m
 
i
injector_fw
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.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