LLM2D
连接大型语言模型代理与高性能计算资源
Connecting Large Language Model Agent to High Performance Computing Resource
作者: Heng Ma, Alexander Brace, Carlo Siebenschuh, Greg Pauloski, Ian Foster, Arvind Ramanathan
发布日期: 2/19/2025
arXiv ID: oai:arXiv.org:2502.12280v1

摘要

arXiv:2502.12280v1 宣告类型: 交叉 摘要: 大型语言模型代理工作流程使LLM能够调用工具功能,以在特定的科学领域问题上提高性能。要应对大规模的科学研究,它需要访问计算资源和并行计算设置。在这项工作中,我们将在LangChain/LangGraph工具调用设置中实现Parsl,以弥合LLM代理与计算资源之间的差距。我们在Polaris/ALCF的本地工作站和HPC环境中设置了两种工具调用实现并进行了测试。第一个实现使用了Parsl启用的LangChain工具节点,将工具功能并发地排队给Parsl工作者进行并行执行。第二种配置通过将工具功能转换为Parsl集合函数来实现,更适合在超级计算机环境中进行大规模任务。LLM代理工作流程被提示运行分子动力学模拟,模拟了不同的蛋白质结构和模拟条件。这些结果显示,LLM代理工具在可用的计算资源上由Parsl并发管理和执行。