TokenSpeed scheduler originally used a Radix Tree–based memory pool. As new attention patterns became more complex, we deprecated the Radix Tree–based design and implementation, and rebuilt the memory manager with a flat, block-based KV cache architecture.
TokenSpeed now uses a single flat paged pool with heterogeneous views, making it easier to support different attention mechanisms. Similar approaches have also been explored in the community, including @vllm_project's Jenga and LMDeploy’s TurboMind.
This redesign happened around the release of TML’s Inkling, so we were able to support Inkling from day one with the new architecture. We are excited to keep building with the open-source community. 👇