/eqnn-jax

Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.

Primary LanguagePythonCreative Commons Attribution 4.0 InternationalCC-BY-4.0

Stargazers