概览
NequIP 是一个用于构建 E(3)-等变原子间势的软件框架，而不是单一的固定预测模型。它支持从模型训练、编译到用于原子模拟的工作流程。项目 README 还介绍了通过 nequip.net 获取预训练基础势，这些势覆盖元素周期表中的大多数元素，可直接用于模拟，也可针对用户自己的数据集进行适配。
用户可以从自己的训练数据开始，也可以使用提供的预训练势。文档所述工作流程会生成经过训练或编译的势，以供后续使用，包括通过 ASE calculator 和 LAMMPS 集成使用。来源摘录说明了这些工作流程中的用途，但未说明数据集模式、所需标签或确切的预测接口；这些细节需要查阅链接指南。软件包入口涵盖训练、打包、编译，以及为 LAMMPS ML-IAP 集成做准备。
对于计算工作流程，NequIP 文档介绍了编译后的训练与推理、多 GPU 训练，以及通过 OpenEquivariance 和 CuEquivariance 实现的 GPU 内核加速；其中 CuEquivariance 标记为 alpha。扩展包支持自定义架构和训练方法，Allegro 则被列为一个独立扩展，用于实现严格局域的等变架构。
现有配置需要注意版本兼容性：README 指出 v0.7.0 是一次不向后兼容的更新，并建议旧配置的用户使用 v0.6.2。来源摘录不含数值精度或速度结果，也不能据此确定该软件是否适用于特定化学体系。仓库的 MIT 代码许可证不应被假定为同样规定了单独分发的模型权重或数据集的使用条款。
主要功能
- 用于构建 E(3)-等变原子间势的框架，支持通过扩展自定义架构和训练技术。
- 通过 nequip.net 分发的预训练基础势，文档介绍了下载、编译、运行和微调这些势的工作流程。
- 编译后的训练和推理工作流程，以及多 GPU 训练。
- 与 OpenEquivariance 和 CuEquivariance 集成的 GPU 内核加速；README 将 CuEquivariance 标记为 alpha。
- ASE calculator 集成，以及通过 pair_nequip_allegro pair styles 和 ML-IAP 集成提供的 LAMMPS 支持。
使用场景
- 建议的评估步骤：在将基础势用于更广泛的模拟之前，先用目标材料体系中的代表性结构进行评估。
- 建议的评估步骤：使用项目专属数据集微调预训练势，并评估其是否适用于预期的化学领域。
- 建议的评估步骤：将训练或编译后的势接入 ASE 或 LAMMPS 工作流程，并检查相关集成要求。
- 建议的评估步骤：通过扩展包探索自定义架构或训练方法；可将 Allegro 作为此机制的文档示例。