transformers

Update stateful_callbacks state before saving checkpoint

#32115

Merged