feat: support kv-cache checkpoint/restore for reasoning models