MLP models
MLP
MLP (hidden_channels:Sequence[int], activation:flax.linen.module.Module=<PjitFunction of <function selu at 0x107f9b7f0>>, kernel_init:Union[jax.nn.initializers.Initializer, collections.abc.Callable[...,Any]]=<function init>, use_bias:bool=True, layer_norm:bool=False, parent:Union[flax.linen.m odule.Module,flax.core.scope.Scope,flax.linen.module._Sentinel,NoneT ype]=<flax.linen.module._Sentinel object at 0x11a1a75e0>, name:Optional[str]=None)
MLP with SELU activation and LeCun normal initialization.