跳转到正文

e3nn

e3nn 是用于构建 E(3) 等变神经网络的基础库,提供不可约表示、张量积和球谐函数等操作。

条目更新 ·

概览

它为三维数据和原子模型的网络开发提供数学组件。化学 AI 相关性由已核验的 MACE 官方仓库中的 e3nn 后端说明支持,不能将其直接当作预训练原子间势。

限制

需要自行构建或选择下游模型。README 明确提示主分支不稳定以及版本间可能存在破坏性变更。

主要功能

  • 等变网络数学组件
  • 张量积与球谐函数

使用场景

  • 研究三维等变模型
  • 开发原子模型的网络部件

使用方式

先准备 PyTorch,按官方指南安装 e3nn。从表示和张量积示例开始,为下游模型固定经过核对的版本。

相关指南