OptArrayΒΆ

Optimizer state for an array.

Abstract Signature:

OptArray(value)

PyTorch

API: β€”
Strategy: Custom / Partial

Keras

API: β€”
Strategy: Custom / Partial

TensorFlow

API: β€”
Strategy: Custom / Partial

Apple MLX

API: β€”
Strategy: Custom / Partial

Flax NNX

API: flax.nnx.OptArray
Strategy: Direct Mapping

PaxML / Praxis

API: β€”
Strategy: Custom / Partial