项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-09-22 · 当前星标 ★ 771
技术标签
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- NVIDIA GPU 40G+
- 两个模型的 checkpoint
安装与启动步骤
-
1建环境
让两个模型直接通过 KV-Cache 通信。
conda create -n rosetta python=3.10 -y && conda activate rosetta -
2装本体与训练依赖
训练与评测是分开的 extra。
pip install -e . pip install -e ".[training,evaluation]" -
3准备两个 checkpoint
至少要两个模型的权重。
python script/playground/live_chat_example.py --checkpoint_dir path/to/ckpt1 path/to/ckpt2 -
4跑评测
用统一评测器对齐口径。
python script/evaluation/unified_evaluator.py --config recipe/eval_recipe/unified_eval.yaml
如何确认成功
双模型通信的准确率高于纯文本传递的基线。
注意事项
- 思路是跳过「解码成文字再编码」,省掉信息损失。
- 训练成本高,先用作者发布的 checkpoint 验证。
- 两个模型的结构要能对齐,否则投影层学不动。
核心亮点
- 用 KV-Cache 直接通信替代文本中转,思路新颖且直击多模型协作的 token 与延迟开销
- 提供 ICLR'26 论文的官方实现,可复现论文实验,研究价值明确
- 技术栈聚焦 kv-cache 与 multi-agent,代码方向清晰,便于二次实验
不足之处
- 早期项目,文档与使用示例可能不完整
- 依赖特定模型与缓存格式,跨模型泛化与工程落地仍需验证
适用场景
- 多智能体协作推理中减少模型间通信开销
- 研究 LLM 间语义级信息传递与缓存复用
- 推理加速场景下探索非文本形式的模型接口
替代项目
AutoGen、LangGraph
上一篇:streamlit-webrtc
下一篇:ai-agents
同类项目推荐
langchain4j-community
开源
按需引入社区模块,让 Java 应用快速接上各家大模型与向量库
LangChain4j integrations that are maintained by the community
generative-ai-project-template
开源
一键生成AI项目骨架,省去配置烦恼,专注业务开发。
Template for a new generative ai project using uv, nicegui, fastapi, llms (cloud & l···
PiDeck
开源
把散落的 AI 会话统一收进桌面,随时恢复继续干
PiDeck 是一个开源的桌面工作台,用于在本地项目目录中统一管理 PI | DSH Agent 会话,并支···
RAGLAB
开源
像搭积木一样对比RAG算法,论文复现不再头疼
[EMNLP 2024: Demo Oral] RAGLAB: A Modular and Research-Oriented Unified Framework fo···
miniagent
开源
几十行代码跑通 ReAct,让 LLM 自己调工具干活
A lightweight intelligent agent framework implementing the complete ReAct pattern
unionml
开源
用装饰器把模型一键变成微服务,快速上线 API
UnionML: the easiest way to build and deploy machine learning microservices
openai
开源
Flutter 应用一行接入大模型,聊天语音微调全支持
The unofficial OpenAI SDK for Dart & Flutter. Full API coverage + OpenAI-compatible ···
openai-agents-go
开源
用 Go 快速搭多智能体工作流,接 OpenAI 模型
A lightweight, powerful framework for multi-agent workflows in Go
ToolRegistry
开源
一套工具代码,无缝对接所有主流大模型
ToolRegistry: A Protocol-Agnostic Tool Management Library for Function-Calling LLMs ···
