MLP models


source

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.