Fully document the internal state of the NeuralGCM models on sigma coordinates

[shoyer]

This is important for users who want to manipulate it directly (e.g., for https://github.com/google-research/neuralgcm/issues/8).