ml_switcheroo.plugins
=====================

.. py:module:: ml_switcheroo.plugins

.. autoapi-nested-parse::

   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
----------

.. toctree::
   :maxdepth: 1

   /api/ml_switcheroo/plugins/arg_packing/index
   /api/ml_switcheroo/plugins/attention_packing/index
   /api/ml_switcheroo/plugins/batch_norm/index
   /api/ml_switcheroo/plugins/casting/index
   /api/ml_switcheroo/plugins/checkpoint_keys/index
   /api/ml_switcheroo/plugins/clipping/index
   /api/ml_switcheroo/plugins/context_to_function_wrap/index
   /api/ml_switcheroo/plugins/data_loader/index
   /api/ml_switcheroo/plugins/decompositions/index
   /api/ml_switcheroo/plugins/device_allocator/index
   /api/ml_switcheroo/plugins/device_checks/index
   /api/ml_switcheroo/plugins/einsum/index
   /api/ml_switcheroo/plugins/flatten/index
   /api/ml_switcheroo/plugins/gather/index
   /api/ml_switcheroo/plugins/inplace_unroll/index
   /api/ml_switcheroo/plugins/io_handler/index
   /api/ml_switcheroo/plugins/keras_sequential/index
   /api/ml_switcheroo/plugins/loop_unroll/index
   /api/ml_switcheroo/plugins/loss_wrapper/index
   /api/ml_switcheroo/plugins/method_property/index
   /api/ml_switcheroo/plugins/mlx_extras/index
   /api/ml_switcheroo/plugins/mlx_optimizers/index
   /api/ml_switcheroo/plugins/nnx_to_torch_params/index
   /api/ml_switcheroo/plugins/optimizer_step/index
   /api/ml_switcheroo/plugins/padding/index
   /api/ml_switcheroo/plugins/reshape/index
   /api/ml_switcheroo/plugins/rng_threading/index
   /api/ml_switcheroo/plugins/scatter/index
   /api/ml_switcheroo/plugins/schedulers/index
   /api/ml_switcheroo/plugins/shape_packing/index
   /api/ml_switcheroo/plugins/state_container/index
   /api/ml_switcheroo/plugins/state_flag_injection/index
   /api/ml_switcheroo/plugins/static_unroll/index
   /api/ml_switcheroo/plugins/tf_data_loader/index
   /api/ml_switcheroo/plugins/topk/index