The 2nd input argument of the created FinalizerProcess.next expects a value
matching model_weights_type and its 3rd argument expects value matching
model_weights_type.trainable. The optimizer will be applied to the
trainable model weights only, leaving non_trainable weights unmodified.
The state of the process is the state of the optimizer and the process
returns empty measurements.