optax.ApplyEvery

class optax.ApplyEvery(

count: jax.typing.ArrayLike,

grad_acc: optax.Updates,

)

Contains a counter and a gradient accumulator.