GitHub repository: smsharma/eqnn-jax
Primary language: Python
Description: Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.