enerzyme.models.layers.atom_embedding# Classes BaseAtomEmbedding(max_Za, dim_embedding) NuclearEmbedding(max_Za, dim_embedding[, ...]) RandomAtomEmbedding(max_Za, dim_embedding)