optax.contrib.SplitRealAndImaginaryState

class optax.contrib.SplitRealAndImaginaryState(

inner_state: optax.OptState,

)

Maintains the inner transformation state for split_real_and_imaginary.