enerzyme.models.physnet.interaction.ResidualStack

enerzyme.models.physnet.interaction.ResidualStack(dim_feature: int, num_residual: int, activation_fn: Literal['shifted_softplus', 'swish'] | None = None, activation_params: Dict[Literal['dim_feature', 'initial_alpha', 'initial_beta', 'learnable'], int | float | bool] = {}, initial_weight: Tensor | ndarray | None = None, initial_bias: Tensor | ndarray | None = None, use_bias: bool = True, dropout_rate: float = 0.0) → ResidualStack