def METHOD_NAME(
    embed: Model[Ints2d, Floats2d],
    encode: Model[Padded, Padded],
    predict: Model[Floats2d, Floats2d],
) -> Model[List[Ints1d], Padded]:
    model = chain(
        with_array(embed), with_padded(encode), with_array(predict), list2padded()
    )
    model.set_ref("embed", embed)
    model.set_ref("encode", encode)
    model.set_ref("predict", model.layers[-1])
    return model