optax.TraceState

Contents

optax.TraceState#

class optax.TraceState(
trace: optax.Params,
)[source]#

Holds an aggregation of past updates.