E3J: An Efficient and Open-Source Backend for Euclidean Equivariant Operations on GPU and TPU
DGX agentarXiv:2609.35099v1 Announce Type: new Abstract: We present e3j, a fast Euclid-equivariance backend for geometric deep learning applications with JAX bindings for GPU and TPU. Leveraging both optimized