Optax Wrappers

Status: Coming soon

Planned module: artifex.generative_models.training.optimizers.optax_wrappers

Current owner: artifex.generative_models.training.optimizers.factory

artifex.generative_models.training.optimizers.optax_wrappers is not shipped as a standalone runtime module today. Use the current owner above for the supported runtime surface until a dedicated module exists.

See Training Systems for the current shared training docs.