Building blocksΒΆ Distance Expansions Available Expansions Implementing a new Expansion Envelopes Available Envelopes Implementing a new Envelope Aggregation Base Class Implementations Scaling LocalEnergiesScaler PyTorch Helpers PerElementParameter PerElementEmbedding MLPConfig MLP e3nn Helpers LinearReadOut NonLinearReadOut