ModelAndOptimizer ================= Container combining model and optimizer. **Abstract Signature:** ``ModelAndOptimizer(model, optimizer)`` .. raw:: html

PyTorch

API:
Strategy: Custom / Partial

Keras

API:
Strategy: Custom / Partial

TensorFlow

API:
Strategy: Custom / Partial

Apple MLX

API:
Strategy: Custom / Partial

Flax NNX

API: flax.nnx.ModelAndOptimizer
Strategy: Direct Mapping

PaxML / Praxis

API:
Strategy: Custom / Partial