/e3nn-jax

jax library for E3 Equivariant Neural Networks

Primary LanguagePythonApache License 2.0Apache-2.0

Watchers