Optimizer

Abstract optimizer base class.

PyTorch

API: torch.optim.Optimizer
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.optimizers.Optimizer
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.keras.optimizers.Optimizer
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.optimizers.Optimizer
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.Optimizer
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: —
Strategy: Custom / Partial