optax.MaskedState# class optax.MaskedState( inner_state: Any, )[source]# Maintains inner transform state for masked transformations.
