I think you don’t need checkpoints, if no gradients are calculated.
Checkpoints will trade some memory for compute by re-calculating the intermediate activations needed for the weight updates again. Since apparently no gradients are needed in your computation, you won’t get any benefit from checkpointing.