e3x.nn

Components for building equivariant deep learning architectures.

Modules

e3x.nn.activations	Equivariant implementation of common activation functions.
e3x.nn.features	Utility functions for transforming features.
e3x.nn.functions	Radial and utility functions.
e3x.nn.initializers	Custom initializers.
e3x.nn.modules	Equivariant neural network modules.
e3x.nn.wrappers	Convenience wrappers to simplify usage of other functions.