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.
DeePMD-kit plugin for various graph neural network models
Add a description, image, and links to the nequip topic page so that developers can more easily learn about it.
To associate your repository with the nequip topic, visit your repo's landing page and select "manage topics."