xla
Now that ConfigureDistributedTPU releases all the TPU memory, we do n…
#82
Merged
Merged