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

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

  •  
  •   jingzhou · 2h 40m ago · 109 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   ·   2536 Online   Highest 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 26ms · UTC 16:00 · PVG 00:00 · LAX 09:00 · JFK 12:00
    ♥ Do have faith in what you're doing.