DocumentationKVBM Integrates with Inference frameworks (vLLM, TRTLLM, SGLang) via Connector APIs to influence KV caching behaviour, scheduling, and forward pass execution. There are two components of the interface, Scheduler and Worker. Scheduler(leader) is responsible for the orchestration of KV block offload/onboard, builds metadata specifying transfer data to the workers. It also maintains hooks for handling asynchronous transfer completion. Worker is responsible for reading metadata built by the scheduler(leader), does async onboarding/ offloading at the end of the forward pass.
The following figure shows the typical integration of KVBM with inference frameworks (vLLM used as an example)
vLLM KVBM Integration
Onboarding blocks from Host to Device
Onboarding blocks from Disk to Device
Offloading blocks from Device to Host&Disk
