Adam

Auto-discovered via Consensus (Score: 4.0)

Abstract Signature:

Adam(eps, lr)

PyTorchJAX (Core)Apple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.optim.Adam

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: optax.adam

Strategy: Plugin (optimizer_constructor)

Apple MLX

API: mlx.optimizers.Adam

Strategy: Plugin (mlx_optimizer_init)

Flax NNX

API: optax.adam

Strategy: Plugin (optimizer_constructor)

PaxML / Praxis

API: optax.adam

Strategy: Plugin (optimizer_constructor)