Lamb

Auto-discovered via Consensus (Score: 3.0)

Abstract Signature:

Lamb(b1, b2, eps, eps_root, lr, mask, weight_decay)

JAX (Core)

API: optax.lamb
Strategy: Direct Mapping
Official Docs ↗

Keras

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

TensorFlow

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

Apple MLX

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

Flax NNX

API: optax.lamb
Strategy: Direct Mapping
Official Docs ↗