Fix optimizer docstring

Update Keras version.

Fix `save_model` and `load_model` (keras-team#19924)

Update version number

Restore version string resolution in pip_build.

One time fix to _tf_keras API.

Move _tf_keras directory to the root of the pip package.

Fix JAX vectorize.

Update version number

Fix code style.