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