fix grad_checkpoint feature for torch2.5.x

[Yux1angJi]

In PyTorch version 2.5.x, passing kwargs alongside use_reentrant=True in torch.utils.checkpoint results in an error. To address this, it needs to explicitly set use_reentrant=False to avoid runtime issues.

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

[rwightman]

@Yux1angJi thanks for the PR, I had a todo to sort this out library wide and I just added that now #2394 . I wrapped the torch checkpoint to default the use_reentrant to False and allow an override if needed.