摘要
针对将 PyTorch 模型迁移至 JAX 时手动操作易错且 LLM 难以对齐动态 API 的问题，本文提出一种全自主系统。该系统结合上下文学习与 Oracle 驱动的自调试机制：利用源模型运行生成的动态张量状态作为不可变执行 Oracle，指导测试用例合成；通过自主代理循环执行测试并将错误回溯反馈给 LLM 进行自我修正。实验表明，该方法在数值等价性上显著优于基线，为跨框架迁移提供了可靠蓝图。
AI 推荐理由
论文核心提出基于 Oracle 驱动的自调试闭环，实现 Agent 自我修正与进化。
研究机构
Google, Sunnyvale, CA
Google, Kirkland, WA
Google, Mountain View, CA
论文信息