equinox and e3nn-jax

Equinox is a general-purpose neural network framework for JAX that can be used to implement equivariant architectures, while e3nn-jax is a specialized library providing pre-built E(3)-equivariant building blocks—making them complements that can be used together, with e3nn-jax offering domain-specific components for Equinox users working on geometric deep learning problems.