JAX-Library for building E(3)-equivariant deep learning architectures based on Flax.
conda install conda-forge::e3x