Skip to content
Trending storyDeveloping

KohakuLab open-sources Tokenomos, a rewritten vLLM frontend for small-model inference

1 article1 sourcesince Oct 31Last article 11h ago ·

Overview

AISummary of 1 article

KBlueleaf of KohakuLab says Tokenomos, an open-source rewrite of vLLM's frontend scheduler and KV cache/state manager, cuts the CPU-side overhead that slows continuous batching.

In a post on X, the author reports nearly 3x speedup over vLLM on a small model at batch size 512 on an RTX 5090. The model code is borrowed from vLLM, and the author says the change is a full rewrite of the frontend only.

The post's diagram gives a host-loop step time of 0.12 ms at 8 rows and 0.41 ms at 512 rows, and 0.23 ms of GPU idle time per decode step at 512 rows, against 14.37 ms for vLLM. The author states that the CPU is no longer the bottleneck. These figures are the author's own measurements and have not been independently verified.

Written by AI from the articles below · updated Oct 31, 4:24 PM ET

Check the sources:

Article timeline

The articles in this story. Times are ET.

Oct 31
  1. 琥珀青葉@KohakuLabX
    KohakuLab open-sources Tokenomos, a faster frontend for vLLM small-model inference

    AIKohakuLab's KBlueleaf says Tokenomos is now open source and reduces frontend overhead, including scheduling and KV cache management, in vLLM. The author reports nearly 3x speedup over vLLM on a small model, measured on an RTX 5090, with the model code borrowed from vLLM. The post's diagram shows a host-loop step time of 0.12 ms at 8 rows and 0.41 ms at 512 rows, with GPU idle time of 0.23 ms per decode step at 512 rows, against 14.37 ms for vLLM.

    Image from @KBlueleaf's post

Heat Trend

Not enough continuous observations to show a trend yet.