1 result found Sort:

smsharma
eqnn-jax smsharma
4
31
cc-by-4.0
4
Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.
egnn segnn nequip symmetry n-body-simulation equivariant-neural-networks
Created 2023-05-28
170 commits to main branch, last one 2 months ago