iskra.mlp

class iskra.mlp.NetworkConfig(*, otype: Literal['PyTorchMLP', 'FullyFusedMLP', 'CutlassMLP'], activation: Literal['Softplus', 'Squareplus', 'ReLU', 'LeakyReLU', 'SiLU', 'GELU'], encoding_config: dict[str, str | int] | None, n_neurons: int, n_hidden_layers: int)[SOURCE]