SGD

Auto-generated from jax_code_defs

PyTorch

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

JAX (Core)

API: optax.sgd
Strategy: Plugin (optimizer_constructor)

Apple MLX

API: mlx.optimizers.SGD
Strategy: Plugin (mlx_optimizer_init)