fix: atomic checkpoint save (tmp+os.replace)

opt: cumsum/chunked attention kernels, memory-flat CE, block checkpointing, v2 trainer (proven equivalent, 46 tests)