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。