NVIDIA ALCHEMI工具包加速AI驱动的化学与材料科学模拟

摘要

机器学习原子间势（MLIP）正在变革计算化学与材料科学，但开发者面临缺乏GPU加速的原子模拟工具的问题。现有工具多为CPU中心，在混合工作流中，模型由PyTorch在GPU上加速，而模拟工具为串行CPU，导致效率瓶颈。NVIDIA在Supercomputing 2024上发布了ALCHEMI（AI化学与材料创新实验室），提供领域专用工具包和NIM微服务，包含GPU加速的批量操作，如邻居列表等，旨在弥合PyTorch与原子模拟之间的鸿沟。