mtenn.conversion_utils.e3nn

Representation and Strategy implementations for an E(3)-equivariant model using e3nn. The underlying model that we use is specifically the January 2021 Network model.

Classes

E3NN	mtenn wrapper around the e3nn model.
E3NNLayerNorm	Wrapper class around the torch LayerNorm to match the expected signature in the e3nn model forward pass.