xla

Allow the loop functions to save device specific data (ie, optimizers…

#800

Merged