update checkpointing support for jit #5375

Job	Run time
pytester (ubuntu-22.04, 3.10, latest, core)	12m 56s
pytester (ubuntu-22.04, 3.10, latest, ops)	29m 51s
pytester (ubuntu-22.04, 3.10, nightly, core)	12m 56s
pytester (ubuntu-22.04, 3.10, nightly, ops)	29m 56s
pytester (macOS-14, 3.10, latest, core)	10m 49s
pytester (macOS-14, 3.10, latest, ops)	25m 27s
pytester (macOS-14, 3.10, nightly, core)	9m 22s
pytester (macOS-14, 3.10, nightly, ops)	26m 37s
pytester (windows-2022, 3.10, latest, core)	27m 0s
pytester (windows-2022, 3.10, nightly, core)	25m 43s
pytester (ubuntu-22.04, 3.11, latest)	24m 26s
pytester (ubuntu-22.04, 3.12, latest)	28m 4s
testing-guardian	1s
	4h 23m 8s

Provide feedback