checkpoint (3 optiml tests failing on GPU)