optax.TraceState

Contents

optax.TraceState#

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

Holds an aggregation of past updates.