optax.contrib.COCOBState

class optax.contrib.COCOBState(

init_particles: optax.Updates,

cumulative_gradients: optax.Updates,

scale: optax.Updates,

subgradients: optax.Updates,

reward: optax.Updates,

)

State for COntinuous COin Betting.