ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

oryx.experimental.optimizers.optix.scale_by_adam

Scales updates according to Adam update rules.