smsharma / eqnn-jax

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

Geek Repo

Github PK Tool