OUnke's Stars
google-research/e3x
E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
OUnke/SpookyNet
Reference implementation of "SpookyNet: Learning force fields with electronic degrees of freedom and nonlocal effects"