optax.TraceState# class optax.TraceState(trace: optax.Params)[source]# Holds an aggregation of past updates.