e3x

e3x API.

All functions and classes from the nn, ops, and so3 submodules can be used without specifying their full path within each submodule (to prevent extremely long names). For example, e3x.nn.relu is equivalent to e3x.nn.activations.relu.

e3x.nn

Components for building equivariant deep learning architectures.

e3x.so3

Functions related to elements of \(\mathrm{SO}(3)\).

e3x.ops

Custom basic operations used in e3x.

e3x.config

Global configuration options for e3x.