爱意满满的作品展示区。
jingzhou

做了一个极简的纯 Java 实现的 LLM 推理引擎,支持 PagedAttention,前缀缓存,连续批处理和分块预填充

  •  
  •   jingzhou · 5h 4m ago · 172 views
    做了一个极简的纯 Java 实现的 LLM 推理引擎,实现了现代推理服务系统的核心技术:分页 KV cache ( PagedAttention )+ 前缀缓存( Prefix Caching )+ 连续批处理( Continuous Batching )+ 分块预填充( Chunked Prefill )+ 抢占恢复。全部代码约 1.5K 行,依赖极少,适合入门学习;纯 CPU 推理,用 Java Vector API 做 SIMD 加速。灵感来源于 nano-vllm 。

    目前支持 Qwen3 dense 系列模型——已用 Qwen3-0.6B 和 HuggingFace transformers 的 greedy 输出做过逐 token 完全一致的对齐验证。

    欢迎对大模型推理感兴趣的朋友试用: https://github.com/oujingzhou/vllm4j 欢迎 star
    No Comments Yet
    About   ·   Help   ·   Advertise   ·   Blog   ·   API   ·   FAQ   ·   Solana   ·   1105 Online   Highest 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 25ms · UTC 18:24 · PVG 02:24 · LAX 11:24 · JFK 14:24
    ♥ Do have faith in what you're doing.