Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.
-
Updated
Dec 5, 2024 - Python
Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.
Add a description, image, and links to the segnn topic page so that developers can more easily learn about it.
To associate your repository with the segnn topic, visit your repo's landing page and select "manage topics."