计算化学长期面临精度与速度之间的取舍。基于第一性原理的密度泛函理论精度高但计算昂贵,经典力场速度快但缺乏化学精度。机器学习原子间势能作为桥梁应运而生。NVIDIA ALCHEMI Toolkit 是一套 GPU 加速模拟构建模块,使研究人员能够轻松构建自定义原子级模拟工作流。
ALCHEMI Toolkit 扩展了 Toolkit-Ops 的功能,在加速的化学和材料领域特定内核与深度学习模型之间管理数据流。它提供模块化、PyTorch 原生的框架,支持几何松弛、分子动力学以及多个模拟工作流的组合。
该工具包包含基础 GPU 内核加速器——邻居列表构建、DFT-D3 色散校正和长程静电计算——这些都是分子模拟中的计算热点。ALCHEMI 组合还包括 NIM 微服务,用于 AI 加速的批量原子级模拟。
WeChat
Profile