rstm_r5 with CHECKPOINTING version 2