概览
它为三维数据和原子模型的网络开发提供数学组件。化学 AI 相关性由已核验的 MACE 官方仓库中的 e3nn 后端说明支持,不能将其直接当作预训练原子间势。
限制
需要自行构建或选择下游模型。README 明确提示主分支不稳定以及版本间可能存在破坏性变更。
主要功能
- 等变网络数学组件
- 张量积与球谐函数
使用场景
- 研究三维等变模型
- 开发原子模型的网络部件
使用方式
先准备 PyTorch,按官方指南安装 e3nn。从表示和张量积示例开始,为下游模型固定经过核对的版本。
e3nn 是用于构建 E(3) 等变神经网络的基础库,提供不可约表示、张量积和球谐函数等操作。
条目更新 ·
它为三维数据和原子模型的网络开发提供数学组件。化学 AI 相关性由已核验的 MACE 官方仓库中的 e3nn 后端说明支持,不能将其直接当作预训练原子间势。
限制
需要自行构建或选择下游模型。README 明确提示主分支不稳定以及版本间可能存在破坏性变更。
先准备 PyTorch,按官方指南安装 e3nn。从表示和张量积示例开始,为下游模型固定经过核对的版本。
区分图学习基础设施、等变运算库及面向化学的 DGL 工具包,并从表示、评估和版本边界出发选择。