todo before removing wip

lucidrains · Feb 26, 2021 · 19d6e26 · 19d6e26
1 parent 80da916
commit 19d6e26
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -1,4 +1,4 @@
-## EGNN - Pytorch
+## EGNN - Pytorch (wip)
 
 Implementation of <a href="https://arxiv.org/abs/2102.09844v1">E(n)-Equivariant Graph Neural Networks</a>, in Pytorch. May be eventually used for Alphafold2 replication.
 
@@ -25,6 +25,13 @@ feats, coors = layer1(feats, coors, edges)
 feats, coors = layer2(feats, coors, edges) # (1, 16, 512), (1, 16, 3)
 ```
 
+## Todo
+
+- [ ] add integration with pytorch geometric
+- [ ] add tests for se3 equivariance
+- [ ] add an EGAT (attention flavored variant)
+
+
 ## Citations
 
 ```bibtex