AdafactorΒΆ

Auto-discovered via Consensus (Score: 2.0)

Abstract Signature:

Adafactor(lr, weight_decay)

JAX (Core)

API: optax.adafactor
Strategy: Direct Mapping

Keras

API: keras.optimizers.Adafactor
Strategy: Direct Mapping

TensorFlow

API: tf.keras.optimizers.Adafactor
Strategy: Direct Mapping

Apple MLX

API: mlx.optimizers.Adafactor
Strategy: Direct Mapping

Flax NNX

API: optax.adafactor
Strategy: Direct Mapping