#v1 Publicize Checkpointable and AbstractCheckpointable as defined types. These resolve to Any, but now have a docstring (similar to the treatment of PyTree). Also minor refactoring and restructuring of other type definitions.#3206