Search papers, labs, and topics across Lattice.
This paper introduces execution-state capsules, a novel mechanism for low-latency, small-batch on-device physical-AI serving that allows for complete checkpointing and restoration of execution states, including key-value caches and recurrent states. By utilizing a graph-bound approach, the system achieves sub-millisecond snapshot and restore times on GPUs, significantly enhancing responsiveness for interactive applications like LLM agents and robotics. The results demonstrate that this method can achieve up to 27x speedup over traditional cold prefill methods, highlighting its potential for real-time AI applications.
Achieving up to 27x speedup in execution-state restoration could revolutionize low-latency AI applications, from interactive agents to robotics.
Mainstream LLM serving systems reuse prefix work mainly through paged or radix key-value (KV) caches. This is highly effective for high-throughput, high-concurrency serving, but it manages only one positional fragment of execution state: the KV cache. We study the opposite regime: low-latency, small-batch, on-device physical-AI serving, where interactive LLM agents, speech systems, and robot policies repeatedly branch, reset, interrupt, and re-enter under tight responsiveness budgets. We introduce execution-state capsules, a graph-bound checkpoint and restore mechanism for the complete restorable state at a committed boundary. FlashRT is a white-box, backend-facing kernel runtime whose evaluated NVIDIA CUDA backend runs captured graph plans over contiguous static buffers with no block-table indirection. Because the live state is a closed set of named buffers, a capsule can snapshot, restore, fork, or roll back the whole execution boundary, including KV, recurrent state, convolution state, MTP state, and metadata. This moves reuse from token-addressed KV fragments to graph-bound execution-state boundaries. On an RTX 5090, capsule restore is byte-exact at the stored-state level and token-identical under greedy decode. A KV-only ablation diverges, showing that recurrent state is load-bearing. GPU-resident snapshot and restore are sub-millisecond, and TTFT speedup over cold prefill grows from 3.9x at 2k tokens to 27x at 16k tokens. On Jetson AGX Thor and DGX Spark, the same correctness and structural properties hold. Capsules are not a replacement for high-throughput KV-cache serving; they define a complementary latency-first serving point for explicit execution-state reuse.
