optimize_kl: Improvement list

• Fix sample plot with ground truth
• Option for storing all intermediate latent positions
• String of model and domain to hdf5
• Save nifty random state
• Pseudo code of optimize_kl into doc string
• plottable spelling?
• Check documentaiton on callback (iglobal)