Adafactor

Auto-discovered via Consensus (Score: 2.0)

Abstract Signature:

Adafactor(lr, weight_decay)

JAX (Core)

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

Apple MLX

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

Flax NNX

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