def Params(cls) -> InstantiableParams:
    p = super().Params()

    p.Define('mlp_tpl', layers.linears.MLPBlock.Params(),
             'MLP model parameters.')
    p.Define('softmax_tpl', layers.SingleShardSharedEmbeddingSoftmax.Params(),
             'Input softmax embedding lookup layer.')
    return p