Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.
-
Updated
Jun 12, 2024 - Jupyter Notebook
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."