xla

c7fe0f98 - Add utility functions for distributed checkpointing (#5128)