enerzyme.models.layers.electron_embedding.ResidualMLP

enerzyme.models.layers.electron_embedding.ResidualMLP(dim_embedding: int, num_residual: int, activation_fn: Literal['shifted_softplus', 'swish'], use_bias: bool = True, zero_init: bool = True) → ResidualMLP