Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.
-
Updated
Mar 1, 2025 - Python
#
segnn
Here is 1 public repository matching this topic...
Add this topic to your repo
To associate your repository with the segnn topic, visit your repo's landing page and select "manage topics."