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

1aa9f7a

verified

thefinalboss commited on Aug 24