pytorch-lightning

Fix: AsyncCheckpointIO snapshots tensors to avoid race with parameter mutation

#21079

Merged