mir-group/nequip

NequIP is a code for building E(3)-equivariant interatomic potentials

What it solves

NequIP 提供了一个构建 E(3) 等变分子间势能的框架，这类势能用于模拟系统中原子的能量和受力。它旨在为这些深度等变模型提供高性能的训练和推理，使其在分子模拟中更具数据效率和准确性。

How it works

NequIP 使用 E(3) 等变图神经网络来表示原子系统的几何属性。它被设计为一个灵活的框架，允许用户构建自定义架构并实现新的训练技术。它可以与常见的科学软件如 ASE（Atomic Simulation Environment）和 LAMMPS 集成，用于分子动力学模拟。

Who it’s for

计算化学、材料科学和生物分子模拟领域的研究人员，需要高精度、数据高效的分子间势能来模拟原子尺度的相互作用。

Highlights

  • 编译式训练和推理以提升速度。
  • 支持多 GPU 训练。
  • 通过 OpenEquivariance 与 CuEquivariance 提供 GPU 核心加速。
  • 与 ASE 计算器及 LAMMPS pair styles 集成。
  • 可扩展的架构，允许自定义扩展包，如 Allegro。