equivariance 开源项目推荐（111 个）

gcorso /DiffDock

DistDock的实现：分子对接的扩散步骤、扭曲和转折

neurreps /awesome-neural-geometry

一份精选资源与研究集合，涉及大脑、深度网络及其他领域中表征几何

QUVA-Lab /e2cnn

用于 PyTorch 的 E(2)-等变 CNN 库

HannesStark /EquiBind

EquiBind：用于快速预测小分子与蛋白质结合的三维结构的几何深度学习

QUVA-Lab /escnn

用于 PyTorch 的等变可操纵 CNN 库 https://quva-lab.github.io/escnn/

arneschneuing /DiffSBDD

基于结构的药物设计的欧几里得扩散模型。

lucidrains /egnn-pytorch

在 PyTorch 中实现 E(n)-等变图神经网络

chaitjo /geometric-gnn-dojo

Geometric GNN Dojo 提供统一实现与实验，用于探索几何图神经网络的设计空间（ICML 2023）

igashov /DiffLinker

DiffLinker：用于分子连接子设计的等变三维条件扩散模型

jw9730 /tokengt

[NeurIPS'22] 基于 PyTorch 的 Tokenized Graph Transformer（TokenGT）

lucidrains /se3-transformer-pytorch

在 PyTorch 中实现用于等变自注意力的 SE3-Transformers。该特定仓库旨在与最终的 Alphafold2 复现集成。

lucidrains /equiformer-pytorch

Equiformer 的实现，这是一种达到新 SOTA 的 SE3/E3 等变注意力网络，并被 EquiFold 采用于蛋白质折叠

mfinzi /equivariant-MLP

通过约束求解以编程方式生成等变层的库

gcorso /torsional-diffusion

用于分子构象生成的扭转扩散实现（NeurIPS 2022）

octavian-ganea /equidock_public

EquiDock：用于快速刚性三维蛋白质-蛋白质对接的几何深度学习

abhi1kumar /DEVIANT

[ECCV 2022] DEVIANT 的官方 PyTorch 代码：用于单目三维目标检测的深度等变网络

lucidrains /En-transformer

E(n)-Transformer 的实现，将注意力机制融入 Welling 的 E(n)-等变图神经网络

LPDI-EPFL /DrugFlow

面向从头药物设计的多域分布学习

PASSIONLab /OpenEquivariance

OpenEquivariance：一个快速、开源的 GPU JIT 内核生成器，用于 Clebsch-Gordon 张量积。

boqian-li /ETCH

[ICCV 2025 亮点] ETCH：通过等变紧身度将人体拟合推广至穿衣人体