xla
Do not rely only on the optimizer state to sync the tensors.
#587
Merged

Loading