enerzyme.models.layers.electron_embedding

Functions

ResidualMLP(dim_embedding, num_residual, ...)

Classes

BaseElectronEmbedding(dim_embedding, ...[, ...])
ElectronicEmbedding(dim_embedding, num_residual)	Block for updating atomic features through nonlocal interactions with the electrons.
NonlinearElectronicEmbedding(dim_embedding, ...)	Block for updating atomic features through nonlocal interactions with the electrons.