ml_switcheroo.frameworks.optax_shim
===================================

.. py:module:: ml_switcheroo.frameworks.optax_shim

.. autoapi-nested-parse::

   Optax Scanner Logic.

   This module provides introspection for the Optax library to power the "Ghost Protocol"
   discovery. Optax uses a functional API where optimizers and losses are functions
   returning named tuples or callables, rather than Classes.

   Capabilities:
   1.  Scans `optax.losses` for loss functions.
   2.  Scans root `optax` for optimizer factory functions.
   3.  Filters internal utilities to provide clean Abstract Standard candidates.



Attributes
----------

.. autoapisummary::

   ml_switcheroo.frameworks.optax_shim.optax


Classes
-------

.. autoapisummary::

   ml_switcheroo.frameworks.optax_shim.OptaxScanner


Module Contents
---------------

.. py:data:: optax
   :value: None


.. py:class:: OptaxScanner

   Helper to inspect Optax APIs for the discovery system.


   .. py:method:: scan_optimizers() -> List[ml_switcheroo.core.ghost.GhostRef]
      :staticmethod:


      Scans the Optax public API for optimizer functions.

      Optax optimizers are typically factory functions (e.g., `adam`, `sgd`)
      that return a `GradientTransformation`.



   .. py:method:: scan_losses() -> List[ml_switcheroo.core.ghost.GhostRef]
      :staticmethod:


      Scans `optax.losses` for loss functions.