ml_switcheroo.frameworks.optax_shim

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

optax

Classes

OptaxScanner

Helper to inspect Optax APIs for the discovery system.

Module Contents

ml_switcheroo.frameworks.optax_shim.optax = None

class ml_switcheroo.frameworks.optax_shim.OptaxScanner

Helper to inspect Optax APIs for the discovery system.

static scan_optimizers() → List[ml_switcheroo.core.ghost.GhostRef]

Scans the Optax public API for optimizer functions.

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

static scan_losses() → List[ml_switcheroo.core.ghost.GhostRef]

Scans optax.losses for loss functions.