xla

ebb200b0 - Adapt a few `torch.utils.checkpoint` functions for PyTorch/XLA. (#6178)