update checkpointing support for jit #5321

Job	Run time
pytester (ubuntu-22.04, 3.10, latest, core)	11m 54s
pytester (ubuntu-22.04, 3.10, latest, ops)	16m 39s
pytester (ubuntu-22.04, 3.10, nightly, core)	11m 58s
pytester (ubuntu-22.04, 3.10, nightly, ops)	16m 35s
pytester (macOS-14, 3.10, latest, core)	8m 16s
pytester (macOS-14, 3.10, latest, ops)	21m 21s
pytester (macOS-14, 3.10, nightly, core)	9m 54s
pytester (macOS-14, 3.10, nightly, ops)	16m 37s
pytester (windows-2022, 3.10, latest, core)	16m 15s
pytester (windows-2022, 3.10, nightly, core)	16m 15s
pytester (ubuntu-22.04, 3.11, latest)	16m 34s
pytester (ubuntu-22.04, 3.12, latest)	16m 39s
testing-guardian	1m 15s
	3h 0m 12s

Provide feedback