Can you update the __init__ docstring to make it clear that these should not be specified if the MLP is a layer of a larger MLP?