xla

16498c34 - Add Checkpoint api to use optimization barrier (#3524)

Add Checkpoint api to use optimization barrier (#3524) * Add PyTorch/XLA version of checkpoint with optimization barrier * Add checkpoint code * update test