RMSpropΒΆ

Optimizer that implements the RMSprop algorithm.

Abstract Signature:

RMSprop(learning_rate: float = 0.001, rho: float = 0.9, momentum: float = 0.0, epsilon: float = 1e-07, centered: boolean = False, weight_decay: float)

PyTorch

API: torch.optim.RMSprop
Strategy: Direct Mapping

JAX (Core)

API: optax.rmsprop
Strategy: Direct Mapping

Keras

API: keras.optimizers.RMSprop
Strategy: Direct Mapping

TensorFlow

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

Apple MLX

API: mlx.optimizers.RMSprop
Strategy: Direct Mapping

Flax NNX

API: optax.rmsprop
Strategy: Direct Mapping