Checkpoints also include context.

I also use jj to checkpoint. When working on a change, each time I get to a stable point I squash and start fresh with an empty change.

You can absolutely continue doing that.