Fix bug loading optimizers saved after the Keras repo split.
Fix bug loading optimizers saved after the Keras repo split.

Cleanup the TODO item for save model in keras.

PiperOrigin-RevId: 380246919

keras-team/keras,keras-team/keras