[XGLM] fix gradient checkpointing
# What does this PR do?

Adds the missing import statement for `torch.utils.checkpoint`

Fixes #15422 
_The documentation is not available anymore as the PR was closed or merged._