给 LLM 推理加速的 KV Cache 层,可对接 vLLM 等框架,复用和缓存键值对来降低显存占用、提升吞吐,支持 CUDA 和 ROCm。
RWKV 是一种兼具 RNN 与 Transformer 优势的语言模型架构,支持像 GPT 一样并行训练,推理时线性时间、常量显存、无限上下文,无需 KV 缓存。
用 PyTorch 从零实现的轻量级 vLLM,代码精简易读,适合学习大模型推理引擎的调度、KV Cache 和 PagedAttention 等核心机制。
高性能内存键值数据库,兼作缓存、消息队列和向量检索引擎,帮实时数据应用把读写延迟压到毫秒级。
在工具输出、日志、文件和 RAG 片段进入 LLM 前先做压缩,编码 Agent 省约 20% token,JSON 省 60-95%,答案不变。提供库、代理和 MCP 服务器三种用法。
面向大语言模型的高吞吐、低显存推理与服务引擎,通过 PagedAttention 等优化显著提升并发吞吐并降低显存占用,适合部署在线推理服务。
从零用 PyTorch 一步步实现类 ChatGPT 大语言模型,覆盖注意力机制、预训练、指令微调全流程,适合边跑代码边理解 LLM 内部原理。
快速且省显存的精确注意力实现,支持训练和推理,能显著加速 Transformer 长序列计算。
用 Python API 定义大模型,并借助 TensorRT 在 NVIDIA GPU 上做极致优化推理,还提供 Python/C++ 运行时来编排高性能服务。
…ms) for counterparty screening. Local Inference & Cache Benchmarking - KV Cache Scaling Harness ( benchmark kv cache.py ) : Added reproducible benchmark harness measuring
…ustom operators TRT Attention, TRT MoE, and TRT KVCacheUpdate - Removed deprecated APIs - IParser::supportsModel and IParser::parseWithWeightDescriptors
…establishes clean extension points for upcoming KVCache-based injection predictors while keeping the current prefill queue behavior stable and reproducible.
…9.0 - Support SGLang v0.5.10 Bug Fixes - Fix RadixCache Patch version issue in SGLang ( 296) - Fix bug when enable APC with GPT-OSS ( 292) - Fix async scheduler support o
… when TP is enabled ( 219) - Support flashinfer kvcache shape for vLLM ( 216) - Fix exceptions in vLLM prefix-cache-related APIs ( 212) - Export missing autopatch env ( 1
- Fix KV cache allocation bugs - Enhance examples - Add GSMK8 benchmark - Update docs and dockers