ml_switcheroo.plugins

Plugins Package.

Automatically discovers and registers all plugin modules within this package. This ensures that adding a new file (e.g., my_new_op.py) automatically registers its hooks without manual edits to this file.

Submodules

• 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