enerzyme.models.spookynet.interaction.ResidualStack

enerzyme.models.spookynet.interaction.ResidualStack(dim_embedding: int, num_residual: int, activation_fn: Literal['shifted_softplus', 'swish'], use_bias: bool = True, zero_init: bool = True) → ResidualStack