/e3x

E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.

Primary LanguagePythonApache License 2.0Apache-2.0

Issues