Adamw

Auto-discovered via Consensus (Score: 4.0)

Abstract Signature:

Adamw(eps, lr, weight_decay)

JAX (Core)Apple MLXFlax NNX

JAX (Core)

API: optax.adamw

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.optimizers.optimizers.AdamW

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: optax.adamw

Strategy: Direct Mapping

Official Docs ↗