xla

591c3979 - Reenable the distributed checkpointing test (#8424)